Evolving Error States: Failure-Aware Progressive Repair for Ultrasound Lesion Segmentation
Organizations: Central South University · Yangtze University · Chongqing University · Peking University ·
Abstract
Reliability under sparse and heterogeneous failures remains a fundamental challenge for medical image segmentation. High average accuracy can conceal a small set of structurally distinct and clinically consequential errors. Existing post-hoc correction methods alleviate this problem, but typically estimate false-positive and false-negative corrections from the same fixed prediction. This ignores the dynamic evolution of error states and limits the correction of complex cases. Inspired by iterative error feedback in structured prediction, we propose Failure-Aware Progressive Repair (FAPR). FAPR represents the current segmentation mask as a dynamic failure state and models each repair operation as a state-transition operator. Each accepted correction forms a new prediction state for subsequent error diagnosis and repair, enabling later operations to adapt to preceding changes. Conditional routing selectively activates necessary state transitions, while failure replay exposes the model to rare error states. By keeping the base segmentor frozen, FAPR preserves its established segmentation capability while improving difficult cases. Across three public ultrasound lesion segmentation benchmarks, FAPR improves mean DSC by 1.52%. On the very-hard subsets of BUSI and TN3K, the average gain reaches 13.77%.
Figures & tables
| BUSI | BUSIS | TN3K | |||||||||||
|---|---|---|---|---|---|---|---|---|---|---|---|---|---|
| Method | Params (M) | DSC | mIoU | Prec. | Rec. | DSC | mIoU | Prec. | Rec. | DSC | mIoU | Prec. | Rec. |
| U-Net (Ronneberger et al., 2015) | 31.04 | 80.91 | 72.16 | 81.84 | 84.77 | 91.18 | 84.89 | 93.02 | 90.88 | 80.48 | 71.04 | 82.26 | 84.53 |
| AttUNet (Oktay et al., 2018) | 34.88 | 76.62 | 68.09 | 79.72 | 78.44 | 91.04 | 84.65 | 93.04 | 90.66 | 77.80 | 67.86 | 74.94 | 87.04 |
| SegRefiner (Wang et al., 2023) | 124.96 | 81.34 | 72.46 | 83.35 | 83.71 | 92.63 | 86.80 | 93.33 | 92.76 | 81.91 | 73.23 | 85.87 | 84.38 |
| CMUNet (Tang et al., 2023) | 49.93 | 79.95 | 71.68 | 84.13 | 81.45 | 91.43 | 85.28 | 92.86 | 91.58 | 79.86 | 70.15 | 78.35 | 85.19 |
| MSAGHNet (Zhu et al., 2025) | 26.61 | 81.44 | 72.53 | 84.04 | 81.78 | 91.36 | 84.76 | 92.88 | 90.94 | 81.01 | 71.35 | 79.83 | 87.55 |
| Dataset | Method | Very hard | Hard | Easy |
|---|---|---|---|---|
| BUSI | Stage 1 | 17.04 | 42.62 | 89.18 |
| SegRefiner | 17.22 | 42.54 | 89.32 | |
| CausalBridgeNet | 17.20 | 43.80 | 89.58 | |
| FAPR | 35.78 | 53.69 | 90.11 | |
| TN3K | Stage 1 | 25.97 | 42.85 | 89.60 |
| SegRefiner | 25.93 | 42.82 | 89.68 |
| Configuration | Overall | Easy | Hard | Very hard |
|---|---|---|---|---|
| FAPR | 83.78 | 90.11 | 53.69 | 35.78 |
| w/o Gross-miss | 83.28 | 90.03 | 50.49 | 30.40 |
| w/o FN Expansion | 82.48 | 89.77 | 47.00 | 21.58 |
| w/o FP Suppression | 82.31 | 89.84 | 45.67 | 21.66 |
| w/o Failure Replay | 82.38 | 89.90 | 45.83 | 18.95 |
| w/o Controlled Corruption | 82.40 | 89.89 | 45.97 | 19.51 |
| Configuration | Total | Train. | DSC | mIoU | Hard | Very hard |
|---|---|---|---|---|---|---|
| (M) | (M) | DSC | DSC | |||
| U-KAN-Large (GB16) | 25.36 | 25.36 | 78.76 | 68.85 | 47.36 | 28.96 |
| Shared Refiner | 17.32 | 10.70 | 82.69 | 74.17 | 48.01 | 27.77 |
| Parallel Expert Fusion | 17.61 | 10.99 | 82.99 | 74.41 | 48.84 | 29.21 |
| Static- Cascade | 17.37 | 10.75 | 82.98 | 74.45 | 50.88 | 33.61 |
| Reverse Repair Order | 17.37 | 10.75 | 82.55 | 73.92 | 46.02 | 21.83 |
| Dataset | Method | DSC | mIoU | Hard | Very hard |
|---|---|---|---|---|---|
| BUSI | U-Net Stage1 | 80.91 | 72.16 | 38.36 | 16.38 |
| U-Net + FAPR | 82.86 | 74.71 | 46.47 | 30.37 | |
| TN3K | U-Net Stage1 | 80.48 | 71.04 | 42.38 | 21.83 |
| U-Net + FAPR | 82.55 | 73.22 | 52.19 | 37.83 |
Medical Image Segmentation
Medical image segmentation delineates organs, lesions, and other structures in scans to support measurement, diagnosis, treatment planning, and research.
See all 788 papers