ErrorRoute: Budgeted Utility-Aware Resolution Routingfor 3D Medical Image Segmentation
Chen, J.; Chen, W.
Show abstract
Three-dimensional medical image segmentation is unusually heterogeneous in its computational needs: large homogeneous organ interiors are often resolved at coarse scale, whereas small organs, lesions, thin vessels, and ambiguous boundaries benefit from native-resolution processing. Yet modern 3D segmentation networks allocate nearly uniform computation across the entire volume. Existing dynamic-resolution or coarse-to fine methods reduce cost, but typically route regions using hand-crafted image complexity, prediction confidence, or a fixed crop, none of which directly estimates whether spending additional computation on a region will improve the segmentation. We introduce ERRORROUTE, a budgeted inference framework that learns the marginal refinement utility of candidate 3D blocks. A coarse branch first predicts global low-resolution semantics. During training, we estimate for each candidate block the loss reduction achievable by native resolution refinement and supervise a utility estimator to predict this gain before refinement. A differentiable budgeted selector then allocates a fixed compute budget to blocks with the highest expected utility per unit cost, while unselected regions remain on the efficient global path. Selected blocks are refined locally and merged through residual cross-scale fusion. Across six 3D CT/MRI benchmarks, ERRORROUTE achieves a mean Dice of 90.5% at 128 GFLOPs, improving the accuracy-compute frontier over dense 3D baselines and fixed coarse-to-fine systems. At a 30% routing budget, the learned utility estimator recovers 83% of the blocks selected by an oracle router, with the largest allocation concentrated on lesions, small structures, and complex boundaries. These results suggest that efficient medical segmentation benefits from learning where additional computation is useful, rather than merely where the model is uncertain.
Matching journals
The top 7 journals account for 50% of the predicted probability mass.
Similar papers in this journal
Similar papers in this journal
- Generative AI Enables Medical Image Segmentation in Ultra Low-Data Regimes 95%
- Weakly supervised classification of rare aortic valve malformations using unlabeled cardiac MRI sequences 93%
- Segmenting functional tissue units across human organs using community-driven development of generalizable machine learning algorithms 92%
Similar papers in this journal
- Phase Recognition in Contrast-Enhanced CT Scans based on Deep Learning and Random Sampling 91%
- Necessity and Impact of Specialization of Large Foundation Model for Medical Segmentation Tasks 91%
- SCU-Net: A deep learning method for segmentation and quantification of breast arterial calcifications on mammograms 89%
Similar papers in this journal
- Encrypted federated learning for secure decentralized collaboration in cancer image analysis 92%
- Track-To-Learn: A general framework for tractography with deep reinforcement learning 92%
- A Framework for Falsifiable Explanations of Machine Learning Models with an Application in Computational Pathology 92%
"Similar papers" are the closest papers from that journal in the model's embedding space. They show what the match is built on, but the ranking comes mostly from a classifier over the whole training set, not from these examples alone.