An increase in stormwater frequency following the rapid development of urbanization has drawn attention to the mitigating strategies in recent decades. For the first time, the present study aims to conduct a local rehabilitation in stormwater collecting systems by (i) detecting the critical nodes along with the canal network and (ii) redesigning the critical canal reaches using Ant Colony Optimization (ACO) to create maximum capacity for flood discharge with minimum reconstruction cost while considering the probability of exceedance of the flood as a constraint. Hence, using the SWMM model, the flow in the collection system was simulated, and the inundation points in the study area in the eastern Tehran metropolis were determined. After determining the critical points, the hydraulic stimulation model for the selected canal flows was developed using HEC-RAS software to accurately simulate each critical bridge's flow. Then, the optimal parameters for the canal bed width and canal depth were obtained using ACO and defining a probability objective function using the flood probability exceedance as the redesign constraint. The results from the optimizer were compared with those of the LINGO nonlinear model. Finally, the operational performance of the redesigned system was evaluated using the optimal selected parameters. The results showed that in redesigning the studied canals, the two widening and deepening options are needed to obtain a discharge with sufficient flow capacity in various return periods (RPs). The optimization results for the first to third critical sections for a design discharge with a 100-year RPs showed that the calculated cost was 19.765(*106), 13.327(*106), and 43.139(*106) IR Rials (1USD = 202000IRR), respectively. For the selected sections, the optimal bed width is 6.97, 8.97, and 10.93 meters, and the optimal depth is 3.68, 4.81, and 4.04 meters, respectively. The results indicate that the local modification in the eastern flood control canal adequately improved inundation problem reduction in various RPs – i.e., for a 10-year RP, the number of node flooding dropped from 4 to zero, the inundated area from 17 percent to zero, and the overflow volume from (10–45) to zero. It also reduced overflow volume from (30–65), (43–74), and (70–92) in the status quo to (4–12), (11–27), and (24–36) percent for precipitations with 25, 50 and 100-year RPs, respectively.