Optimizing Neural Network Architecture for Medical Image Segmentation Using Monte Carlo Tree Search
Open accessAbstract
This paper proposes a novel medical image segmentation framework, MNAS-Unet, which combines Monte Carlo Tree Search (MCTS) and Neural Architecture Search (NAS). MNAS-Unet dynamically explores promising network architectures through MCTS, significantly enhancing the efficiency and accuracy of architecture search. It also optimizes the DownSC and UpSC unit structures, enabling fast and precise model adjustments. Experimental results demonstrate that MNAS-Unet outperforms NAS-Unet and other state-of-the-art models in segmentation accuracy on several medical image datasets, including PROMISE12, Ultrasound Nerve, and CHAOS. Furthermore, compared with NAS-Unet, MNAS-Unet reduces the architecture search budget by 54% (early stopping at 139 epochs versus 300 epochs under the same search setting), while achieving a lightweight model with only 0.6M parameters and lower GPU memory consumption, which further improves its practical applicability. These results suggest that MNAS-Unet can improve search efficiency while maintaining competitive segmentation accuracy under practical resource constraints.
AI snapshot
Written from the paper’s full text.
TL;DR
MNAS-Unet integrates Monte Carlo Tree Search with Neural Architecture Search to discover U-shaped cell architectures for medical image segmentation using a tailored search space of 6 Down, 4 Up, and 6 Normal POs. It reports higher DSC/mIoU than NAS-Unet and other baselines on PROMISE12, Ultrasound Nerve, and CHAOS while reducing search epochs from 300 to 139 and using 0.6M parameters.
Key findings
- MNAS-Unet achieved the best reported segmentation scores in Table 2: PROMISE12 mIoU 0.573 / DSC 0.728, Ultrasound mIoU 0.727 / DSC 0.842, and CHAOS mIoU 0.933 / DSC 0.966, compared with NAS-Unet at 0.561 / 0.719, 0.714 / 0.833, and 0.927 / 0.962 respectively.
- Architecture search cost was reduced by approximately 54%: MNAS-Unet early-stopped at 139 epochs versus 300 epochs for NAS-Unet under the same search setting.
- The model is lightweight at only 0.6M parameters and reports lower peak GPU memory than NAS-Unet: 5.8GB vs 6.5GB on PROMISE12, 6.3GB vs 6.8GB on Ultrasound, and 7.0GB vs 7.5GB on CHAOS.
- The search space contains 6 Down POs, 4 Up POs, and 6 Normal POs, with 8 cells and 4 intermediate nodes per cell.
Methods
- Design: MCTS-guided NAS on a U-shaped backbone with DownSC and UpSC cells represented as a DAG; MCTS uses selection, expansion, simulation, and backpropagation with UCB1, and the search space includes 6 Down POs, 4 Up POs, and 6 Normal POs.
- Data: PASCAL VOC 2012 was used for proxy architecture search; medical evaluation used PROMISE12 (50 training cases, around 1,250 labeled images, 256×256), Ultrasound Nerve Segmentation (5,635 training and 5,508 test images, 580×420), and CHAOS (CT/MRI abdominal organ and liver segmentation). Baselines included U-Net, FC-DenseNet, NAS-Unet, DC-UNet, DCSAU-Net, MNet, MedNeXt, and ResTransUNet.
- Analysis: Models were implemented in PyTorch on an NVIDIA RTX 4090 with Python 3.8, PyTorch 1.11.0, and CUDA 11.3; training used batch size 16, 300 epochs, Dice Loss, AdaBound for weights (lr 0.0005, weight decay 0.0005), and Adam for architecture parameters (lr 0.0001, weight decay 0.005). Metrics were DSC, mIoU, validation loss, pixel accuracy, and IoU.
Limitations
- Paper-stated: future work will focus on enhancing the interpretability of MNAS-Unet, particularly explaining decision-making processes and identifying key features relevant to clinical diagnoses.
- Evident from text: architecture search was performed on PASCAL VOC 2012 as a proxy and then adapted to medical datasets, rather than searching directly on each medical dataset.
- Evident from text: Table 2 reports single-run wall-clock times and no variance or statistical significance tests; the Ultrasound Nerve dataset is also noted to contain inconsistent or conflicting labels, with around 60% of training images lacking the BP region.
Model-written summary. Check it against the paper before relying on it.
Integrity signals
Registry
Cites
Statistics
Claims the paper makes
10 of 13 cited sources checked · 5 supported · 0 flagged · accuracy not yet measured
- Findingown data§ ABSTRACT · p. 3
MNAS-Unet achieves a lightweight model with only 0.6M parameters.
“while achieving a lightweight model with only 0.6M parameters and lower GPU memory consumption”
- Also stated in: § Optimizing Neural Network Architecture for Medical Image Segmentation Using Monte Carlo Tree Search · p. 2 · plain
“Lightweight model with only 0.6M parameters and lower GPU memory consumption”
Quote in the paperSubstantial claim - Also stated in: § Optimizing Neural Network Architecture for Medical Image Segmentation Using Monte Carlo Tree Search · p. 2 · plain
- Findingown data§ ABSTRACT · p. 3
MNAS-Unet outperforms NAS-Unet and other state-of-the-art models in segmentation accuracy on several medical image datasets, including PROMISE12, Ultrasound Nerve, and CHAOS.
“Experimental results demonstrate that MNAS-Unet outperforms NAS-Unet and other state-of-the-art models in segmentation accuracy on several medical image datasets, including PROMISE12, Ultrasound Nerve, and CHAOS.”
- Also stated in: § Optimizing Neural Network Architecture for Medical Image Segmentation Using Monte Carlo Tree Search · p. 2 · plain
“Achieved superior performance on multiple medical datasets (PROMISE12, CHAOS, Ultrasound)”
Quote in the paperSubstantial claim - Also stated in: § Optimizing Neural Network Architecture for Medical Image Segmentation Using Monte Carlo Tree Search · p. 2 · plain
- Findingown data§ ABSTRACT · p. 3
MNAS-Unet has lower GPU memory consumption than NAS-Unet.
“while achieving a lightweight model with only 0.6M parameters and lower GPU memory consumption”
- Also stated in: § Optimizing Neural Network Architecture for Medical Image Segmentation Using Monte Carlo Tree Search · p. 2 · plain
“Lightweight model with only 0.6M parameters and lower GPU memory consumption”
Quote in the paperSubstantial claim - Also stated in: § Optimizing Neural Network Architecture for Medical Image Segmentation Using Monte Carlo Tree Search · p. 2 · plain
- Findingown data§ ABSTRACT · p. 3
Compared with NAS-Unet, MNAS-Unet reduces the architecture search budget by 54% (early stopping at 139 epochs versus 300 epochs under the same search setting).
“compared with NASUnet, MNAS-Unet reduces the architecture search budget by 54% (early stopping at 139 epochs versus 300 epochs under the same search setting)”
- Also stated in: § Optimizing Neural Network Architecture for Medical Image Segmentation Using Monte Carlo Tree Search · p. 2 · hedged
“Reduced architecture search cost by ∼ 54% (139 vs. 300 search epochs) compared to NAS-Unet”
Quote in the paperSubstantial claim - Also stated in: § Optimizing Neural Network Architecture for Medical Image Segmentation Using Monte Carlo Tree Search · p. 2 · hedged
- Findingown data§ 4.5. Experimental Results · p. 8
Under the same search setting, NAS-Unet runs for the full 300 epochs.
“Under the same search setting, NAS-Unet runs for the full 300 epochs, whereas MNAS-Unet converges earlier (Fig. 7 vs. Fig. 8).”
Quote in the paperSubstantial claim
- Findingown data§ 4.5. Experimental Results · p. 8
MNAS-Unet's architecture search terminated early at 139 epochs due to the patience limit.
“The early termination of MNAS-Unet at 139 epochs due to the patience limit indicates improved search efficiency.”
Quote in the paperSubstantial claim - Findingreasoning§ 4.5. Experimental Results · p. 8
MNAS-Unet reduces search cost by approximately 54% in training-epoch budget compared to NAS-Unet.
“In terms of training-epoch budget (and corresponding GPU compute), this reduces the search cost by 1 - 139∕300 ≈ 54% .”
Quote in the paperSubstantial claim- Rests on an assumption: Training-epoch budget is a valid proxy for GPU compute cost in architecture search.
- Findingown data§ 4.5. Experimental Results · p. 8
Under the same search setting, MNAS-Unet converges earlier than NAS-Unet.
“Under the same search setting, NAS-Unet runs for the full 300 epochs, whereas MNAS-Unet converges earlier (Fig. 7 vs. Fig. 8).”
Quote in the paperSubstantial claim - Interpretationown data§ 5. Discussion and Conclusion · p. 9
MCTS improves segmentation accuracy, particularly for medical images with complex structures and textures.
“showcasing the effectiveness of MCTS in improving segmentation accuracy, particularly for medical images with complex structures and textures.”
Quote in the paperSubstantial claim- Rests on an assumption: The selected datasets (PROMISE12, CHAOS, Ultrasound Nerve Segmentation) are representative of medical image segmentation tasks across MRI, CT, and ultrasound modalities.
- Interpretationown data§ 4.5. Experimental Results · p. 8
The early termination of MNAS-Unet at 139 epochs indicates improved search efficiency.
“The early termination of MNAS-Unet at 139 epochs due to the patience limit indicates improved search efficiency.”
Quote in the paperSubstantial claim
- Classified as probably not a substantial claim. 9
- May combine two claims. 10
Classified as probably not a substantial claim.
Study design and conductcitation§ 4.2. Experimental Configurations · p. 7All models were re-implemented in PyTorch to ensure consistency.
“Allmodels were re-implemented in PyTorch [17] to ensure consistency.”
Quote in the paperSubstantial claim1 of 1 sources checked- cites [17] [17] Paszke, A., Gross, S., Massa, F., Lerer, A., Bradbury, J., Chanan, G., Killeen, T., Lin, Z., Gimelshein, N., Antiga, L., Desmaison, A., Kopf, A., Yang, E., DeVito, Z., Raison, M., Tejani, A., Chilamkurthy, S., Steiner, B., Fang, L., Bai, J., Chintala, S., PyTorch: An imperative style, high-performance deep learning library , Advances in Neural Information Processing Systems, 32, 8024-8035, 2019. - source
Probably not a substantial claim: may not pass S3 (load-bearing).
Premisecitation§ ABSTRACT · p. 3U-Net has emerged as a fundamental architecture for medical image segmentation due to its encoder-decoder structure with skip connections.
“U-Net, proposed by Ronneberger et al. (2015), has emerged as a fundamental architecture due to its encoder-decoder structure with skip connections, making it particularly suitable for medical imaging tasks [8].”
Quote in the paperNot a substantial claim1 of 1 sources checked- cites [8] [8] Ronneberger, O., Fischer, P., Brox, T., U-net: Convolutional networks for biomedical image segmentation , Medical image computing and computer-assisted intervention-MICCAI 2015: 18th international conference, Munich, Germany, October 5-9, 2015, proceedings, part III 18, 234-241, 2015. - source
The Network Architecture section (p. 4) describes the contracting and expansive paths and the connections that combine feature maps from both paths. The Conclusion (p. 7) reports strong performance on biomedical segmentation tasks. As the paper introduces U-Net, it does not establish the retrospective claim that U-Net had emerged as a fundamental architecture in the field. (LLM, conf=0.85)
Experimental — accuracy not yet measured
Probably not a substantial claim: may not pass S3 (load-bearing).
Premisecitation§ 3.2. U-Net Model · p. 5The U-Net network features a symmetric encoder-decoder architecture in a distinctive 'U' shape.
“The U-Net network [8], proposed by Ronneberger et al. (2015), features a symmetric encoder-decoder architecture in a distinctive 'U' shape.”
Quote in the paperNot a substantial claim1 of 1 sources checked- cites [8] [8] Ronneberger, O., Fischer, P., Brox, T., U-net: Convolutional networks for biomedical image segmentation , Medical image computing and computer-assisted intervention-MICCAI 2015: 18th international conference, Munich, Germany, October 5-9, 2015, proceedings, part III 18, 234-241, 2015. - source
Probably not a substantial claim: may not pass S3 (load-bearing). May combine two claims.
Premisecitation§ 3.1. Neural Architecture Search · p. 4Neural Architecture Search significantly reduces the manual cost of designing network structures and greatly expands the possibilities for network architecture design.
“Neural Architecture Search (NAS) [21] significantly reduces the manual cost of designing network structures and greatly expands the possibilities for network architecture design.”
Quote in the paperNot a substantial claim1 of 1 sources checked- cites [21] [21] Yang, T., Yang, L., Jin, X., Chen, C., Revisiting training-free nas metrics: An efficient training-based method , Proceedings of the IEEE/CVF Winter Conference on Applications of Computer Vision, 4751-4760, 2023. - source
The Related work section says NAS automatically searches for network structures instead of relying on time-consuming manual design, supporting the claim about reduced manual effort. It also describes exploring architectures, but does not establish that NAS greatly expands the possibilities for architecture design. (LLM, conf=0.78)
Experimental — accuracy not yet measured
Probably not a substantial claim: may not pass S3 (load-bearing).
Premisecitation§ 3.1. Neural Architecture Search · p. 4The algorithm for neural network architecture search typically encompasses three key components: the search space, the search strategy, and the strategy for performance evaluation.
“The algorithm for neural network architecture search typically encompasses three key components: the search space, the search strategy, and the strategy for performance evaluation [31].”
Quote in the paperNot a substantial claim1 of 1 sources checked- cites [31] [31] Miahi, E., Mirroshandel, S.A., Nasr, A., Genetic Neural Architecture Search for automatic assessment of human sperm images , Expert Systems with Applications, 188, 115937, 2022. - source
Probably not a substantial claim: may not pass S3 (load-bearing).
Premisecitation§ ABSTRACT · p. 3NAS-Unet explored differentiable NAS strategies on a U-shaped backbone for medical image segmentation.
“NAS-Unet, proposed in 2019, explored differentiable NAS strategies (e.g., DARTS/ProxylessNAS-style methods) on a U-shaped backbone for medical image segmentation [10].”
Quote in the paperNot a substantial claim1 of 1 sources checked- cites [10] [10] Weng, Y., Zhou, T., Li, Y., Qiu, X., NAS-Unet: Neural architecture search for medical image segmentation , IEEE access, 7, 4424744257, 2019. - source
Classified as probably not a substantial claim.
Premisecitation§ 3.3. Monte Carlo Tree Search (MCTS) · p. 5MCTS balances exploration and exploitation to effectively navigate large decision spaces.
“The Monte Carlo Tree Search (MCTS) algorithm is a strategic search method that balances exploration and exploitation to effectively navigate large decision spaces [25, 34].”
Quote in the paperSubstantial claim1 of 2 sources checked- cites [25] [25] Wang, L., Zhao, Y., Jinnai, Y., Tian, Y., Fonseca, R., Neural architecture search using deep neural networks and monte carlo tree search , Proceedings of the AAAI Conference on Artificial Intelligence, 34(06), 9983-9991, 2020. - source
- cites [34] [34] Elkefi, S., Layeb, S.B., Deep Learning for Solving Loading, Packing, Routing, and Scheduling Problems , Handbook of Formal Optimization, 787-805, 2024. - source
May combine two claims.
Premisecitation§ ABSTRACT · p. 3Manual segmentation of medical images is labor-intensive, time-consuming, and subject to inter-observer variability, making it impractical for large-scale clinical applications.
“However, manual segmentation of medical images is laborintensive, time-consuming, and subject to inter-observer variability, making it impractical for large-scale clinical applications [1, 2].”
Quote in the paperSubstantial claim2 of 2 sources checked- cites [1] [1] Sun, M.W., Troxell, D., Tibshirani, R., Public health factors help explain cross country heterogeneity in excess death during the COVID19 pandemic , Scientific Reports, 13(1), 16196, 2023. - source
The Methods passages discuss modeling COVID-19 excess deaths, and the Results passages discuss predictive performance and feature importance; none addresses manual segmentation of medical images. Searches for medical-image segmentation, inter-observer variability, and the claimed workload and clinical-scale limitations found no relevant passage in our copy. (LLM, conf=0.78)
Experimental — accuracy not yet measured - cites [2] [2] Galbusera, F., Cina, A., Image annotation and curation in radiology: an overview for machine learning practitioners , European Radiology Experimental, 8(1), 11, 2024. - source
The Annotation section says that producing segmentation masks requires significant time and manual labour, especially for datasets containing thousands or millions of images. It does not address inter-observer variability or conclude that manual segmentation is impractical for large-scale clinical applications; its large-scale concern is annotating datasets for machine learning. (LLM, conf=0.86)
Experimental — accuracy not yet measured
Classified as probably not a substantial claim.
Premisecitation§ 3.3. Monte Carlo Tree Search (MCTS) · p. 5MCTS is particularly effective for optimizing decision-making in complex domains such as neural architecture search and games.
“This iterative approach makes MCTS particularly effective for optimizing decision-making in complex domains [35], such as neural architecture search and games, as shown in Fig. 2.”
Quote in the paperSubstantial claim1 of 1 sources checked- cites [35] [35] Kemmerling, M., Lutticke, D., Schmitt, R.H., Beyond games: a systematic review of neural Monte Carlo tree search applications , Applied Intelligence, 54(1), 1020-1046, 2024. - source
The Introduction reports established effectiveness for neural MCTS in games, and §2.3 describes its iterative search process. The Introduction also says that additional planning may improve decision quality, but presents that benefit as a promise rather than an established general result. Our copy does not substantiate neural architecture search as an example (searched: “neural architecture search” and “architecture search”). (LLM, conf=0.83)
Experimental — accuracy not yet measured
May combine two claims.
Findingown data§ 4.5. Experimental Results · p. 8MNAS-Unet achieves the best overall segmentation performance while maintaining lower memory consumption.
“MNAS-Unet achieves the best overall segmentation performance while maintaining lower memory consumption.”
- Also stated in: § 4.5. Experimental Results · p. 9 · plain
“MNAS-Unet | 0.573 | 0.728 | 36min | 5.8GB | 0.727 | 0.842 | 4h41m | 6.3GB | 0.933 | 0.966 | 11h52m | 7.0GB”
Quote in the paperSubstantial claim- Also stated in: § 4.5. Experimental Results · p. 9 · plain
May combine two claims.
Findingown data§ 5. Discussion and Conclusion · p. 9MNAS-Unet outperforms NAS-Unet and other state-of-the-art models in key metrics across multiple datasets, including PROMISE12, Chaos, and ultrasound segmentation.
“Experimental results demonstrate the superiority of MNAS-Unetacrossmultiple datasets, including PROMISE12, Chaos, and ultrasound segmentation. MNAS-Unet outperforms NAS-Unet and other state-of-the-art models in key metrics”
- Also stated in: § 5. Discussion and Conclusion · p. 9 · boosted
“Ourcontributions are the integration of MCTS with NAS for medical image segmentation, the design of a tailored search space (6 Down, 4 Up, and 6 Normal POs), and the demonstration of improved search efficiency and segmentation accuracy across multiple datasets”
Quote in the paperSubstantial claim- Also stated in: § 5. Discussion and Conclusion · p. 9 · boosted
Probably not a substantial claim: may not pass S3 (load-bearing). May combine two claims.
Study design and conductcitation§ 4.2. Experimental Configurations · p. 7The same data augmentation techniques were applied across all models, with additional preprocessing for noisy images.
“The same data augmentation techniques were applied across all models, with additional preprocessing for noisy images [18, 19].”
Quote in the paperNot a substantial claim0 of 2 sources checked- cites [18] [18] Zhou, J., Yang, T., Zhang, W., Underwater vision enhancement technologies: A comprehensive review, challenges, and recent trends , Applied Intelligence, 53(3), 3594-3621, 2023. - source
- cites [19] [19] Lu, H., Li, Y., Uemura, T., Kim, H., Serikawa, S., Low illumination underwater light field images reconstruction using deep convolutional neural networks , Future Generation Computer Systems, 82, 142-148, 2018. - source
May combine two claims.
Interpretationnone given§ ABSTRACT · p. 3MNAS-Unet optimizes the DownSC and UpSC unit structures, enabling fast and precise model adjustments.
“It also optimizes the DownSC and UpSC unit structures, enabling fast and precise model adjustments.”
Quote in the paperSubstantial claimMay combine two claims.
Interpretationown data§ ABSTRACT · p. 3MNAS-Unet can improve search efficiency while maintaining competitive segmentation accuracy under practical resource constraints.
“These results suggest that MNAS-Unet can improve search efficiency while maintaining competitive segmentation accuracy under practical resource constraints.”
- Also stated in: § Optimizing Neural Network Architecture for Medical Image Segmentation Using Monte Carlo Tree Search · p. 2 · plain
“Resource-efficient architecture suitable for deployment under limited GPU memory”
Quote in the paperSubstantial claim- Also stated in: § Optimizing Neural Network Architecture for Medical Image Segmentation Using Monte Carlo Tree Search · p. 2 · plain
May combine two claims.
Interpretationown data§ ABSTRACT · p. 3MNAS-Unet dynamically explores promising network architectures through MCTS, significantly enhancing the efficiency and accuracy of architecture search.
“MNAS-Unet dynamically explores promising network architectures through MCTS, significantly enhancing the efficiency and accuracy of architecture search.”
Quote in the paperSubstantial claimMay combine two claims.
Interpretationown data§ 5. Discussion and Conclusion · p. 9Compared with traditional ProxylessNAS, incorporating MCTS significantly improves architecture search efficiency and optimizes the gradient descent process, accelerating network architecture design and enhancing segmentation accuracy.
“Compared with the traditional ProxylessNAS method, incorporating MCTS significantly improves the architecture search efficiency and optimizes the gradient descent process, thus accelerating the design of the network architecture and enhancing the segmentation accuracy.”
Quote in the paperSubstantial claim- Rests on an assumption: Architectures discovered on PASCAL VOC 2012 can be effectively adapted and transferred to medical image segmentation datasets.
- Rests on an assumption: The selected datasets (PROMISE12, CHAOS, Ultrasound Nerve Segmentation) are representative of medical image segmentation tasks across MRI, CT, and ultrasound modalities.
May combine two claims.
Interpretationown data§ 5. Discussion and Conclusion · p. 9Combining MCTS-guided exploration with a tailored cell search space on a U-shaped backbone improves search efficiency and achieves strong segmentation performance across multiple modalities.
“By combining MCTS-guided exploration with a tailored cell search space on a U-shaped backbone, MNAS-Unet improves search efficiency and achieves strong segmentation performance across multiple modalities.”
Quote in the paperSubstantial claim- Rests on an assumption: Architectures discovered on PASCAL VOC 2012 can be effectively adapted and transferred to medical image segmentation datasets.
The paper relies on this without support.
Manually designing optimal network architectures for specific medical imaging tasks remains a challenging and time-consuming process that typically requires extensive domain expertise and computational resources.
“However, manually designing optimal network architectures for specific medical imaging tasks remains a challenging and timeconsuming process that typically requires extensive domain expertise and computational resources.”
The paper relies on this without support.
Existing NAS methods have high computational requirements that limit practical applicability.
“However, existing NAS methods face several challenges: (1) high computational requirements that limit practical applicability, (2) generic search spaces that do not account for medical imaging characteristics, and (3) insufficient consideration of clinical deployment constraints such as model size and inference speed.”
The paper relies on this without support.
Existing NAS methods use generic search spaces that do not account for medical imaging characteristics.
“However, existing NAS methods face several challenges: (1) high computational requirements that limit practical applicability, (2) generic search spaces that do not account for medical imaging characteristics, and (3) insufficient consideration of clinical deployment constraints such as model size and inference speed.”
The paper relies on this without support.
Existing NAS methods give insufficient consideration to clinical deployment constraints such as model size and inference speed.
“However, existing NAS methods face several challenges: (1) high computational requirements that limit practical applicability, (2) generic search spaces that do not account for medical imaging characteristics, and (3) insufficient consideration of clinical deployment constraints such as model size and inference speed.”
The paper relies on this without support.
Existing U-Net variants require extensive manual design and hyperparameter tuning.
“Despite these advances, existing U-Net variants face several limitations: (1) they require extensive manual design and hyperparameter tuning, (2) their architectures are often suboptimal for specific medical imaging tasks, and (3) they lack adaptability to different medical imaging modalities and anatomical structures.”
The paper relies on this without support.
Existing U-Net variants' architectures are often suboptimal for specific medical imaging tasks.
“Despite these advances, existing U-Net variants face several limitations: (1) they require extensive manual design and hyperparameter tuning, (2) their architectures are often suboptimal for specific medical imaging tasks, and (3) they lack adaptability to different medical imaging modalities and anatomical structures.”
The paper relies on this without support.
Existing U-Net variants lack adaptability to different medical imaging modalities and anatomical structures.
“Despite these advances, existing U-Net variants face several limitations: (1) they require extensive manual design and hyperparameter tuning, (2) their architectures are often suboptimal for specific medical imaging tasks, and (3) they lack adaptability to different medical imaging modalities and anatomical structures.”
The paper relies on this without support.
The integration of NAS with medical image segmentation presents an opportunity to automatically discover architectures tailored to the unique characteristics of medical images.
“The integration of NAS with medical image segmentation presents an opportunity to automatically discover architectures tailored to the unique characteristics of medical images, including their high resolution, anatomical complexity, and clinical requirements for accuracy and interpretability.”
The paper relies on this without support.
Architectures discovered on PASCAL VOC 2012 can be effectively adapted and transferred to medical image segmentation datasets.
“We investigated proxy cell architectures using the more intricate PASCAL VOC 2012 dataset [20] and then adapted these architectures for medical image datasets.”
The paper relies on this without support.
Training-epoch budget is a valid proxy for GPU compute cost in architecture search.
“In terms of training-epoch budget (and corresponding GPU compute), this reduces the search cost by 1 - 139∕300 ≈ 54% .”
The paper relies on this without support.
The selected datasets (PROMISE12, CHAOS, Ultrasound Nerve Segmentation) are representative of medical image segmentation tasks across MRI, CT, and ultrasound modalities.
“To evaluate MNAS-Unet, we use the PASCAL VOC 2012 dataset and three medical imaging datasets: PROMISE12 [14], CHAOS [15], and Ultrasound Nerve Segmentation [16], which include MRI, CT, and ultrasound images.”
Self-check
Percentages, sums and reported differences re-computed from the paper’s own numbers.
Each contradiction between two passages listed here was confirmed by an independent check.
Conflicting names for the model predicting simulation accuracies
MinorThe search-process description says MNAS-Unet learns to generalize performance to unexplored architectures, but later attributes the predicted accuracies used in those simulations to NAS-Unet. The text does not clarify whether the predictor belongs to the proposed method or the baseline.
“while the MNAS-Unet architecture learns the performance characteristics of sampled architectures and generalizes this knowledge to unexplored architectural configurations”
“Pred represents the predicted accuracies in subsequent 𝑘 simulations by NAS-Unet.”
Cites
U-Net: Convolutional Networks for Biomedical Image Segmentation
Ronneberger, O., Fischer, P., & Brox, T. (2015) · Lecture notes in computer science · doi:10.1007/978-3-319-24574-4_28 · Cited by 99,404 · cited 3× · see citing works
“The U-Net network [8], proposed by Ronneberger et al. (2015), features a symmetric encoder-decoder architecture in a distinctive “U” shape.”
InfluentialPyTorch: An Imperative Style, High-Performance Deep Learning Library
Paszke, A., Gross, S., Massa, F., Lerer, A., Bradbury, J., Chanan, G., Killeen, T., Lin, Z., Gimelshein, N., Antiga, L., et al. (2019) · arXiv · doi:10.48550/arxiv.1912.01703 · Cited by 31,473 · cited 1× · see citing works
“All models were re-implemented in PyTorch [17] to ensure consistency.”
Learning Transferable Architectures for Scalable Image Recognition
Zoph, B., Vasudevan, V., Shlens, J., & Le, Q. V. (2018) · IEEE Computer Society Conference on Computer Vision and Pattern Recognition/Proceedings - IEEE Computer Society Conference on Computer Vision and Pattern Recognition/Proceedings · doi:10.1109/cvpr.2018.00907 · Cited by 6,631 · see citing works
Bandit Based Monte-Carlo Planning
Kocsis, L., & Szepesvari, C. (2006) · European Conference on Machine Learning · doi:10.1007/11871842_29 · Cited by 4,073 · cited 1× · see citing works
“In the selection step, MCTS employs the Upper Confidence Bound for Trees (UCT) [26] to determine the most promising nodes, ensuring a balance between exploring new possibilities and exploiting known successful paths.”
DARTS: Differentiable architecture search
Liu, H., Simonyan, K., & Yang, Y. (2018) · arXiv · doi:10.48550/arxiv.1806.09055 · Cited by 3,248 · cited 1× · see citing works
“Traditional NAS methods, such as DARTS [11] and Proxyless-NAS [12], have shown success in general computer vision tasks but face significant challenges when applied to medical image segmentation.”
ProxylessNAS: Direct Neural Architecture Search on Target Task and HardwarePreprint
Cai, H., Zhu, L., & Han, S. (2018) · arXiv (Cornell University) · doi:10.48550/arxiv.1812.00332 · Cited by 1,652 · cited 1× · see citing works
“Traditional NAS methods, such as DARTS [11] and Proxyless-NAS [12], have shown success in general computer vision tasks but face significant challenges when applied to medical image segmentation.”
Auto-DeepLab: Hierarchical Neural Architecture Search for Semantic Image Segmentation
Liu, C., Chen, L.-C., Schroff, F., Adam, H., Hua, W., Yuille, A., & Fei-Fei, L. (2019) · IEEE Computer Society Conference on Computer Vision and Pattern Recognition/Proceedings - IEEE Computer Society Conference on Computer Vision and Pattern Recognition/Proceedings · doi:10.1109/cvpr.2019.00017 · Cited by 1,076 · see citing works
CHAOS Challenge - combined (CT-MR) healthy abdominal organ segmentation
Kavur, A. E., Gezer, N. S., BARIŞ, M. M., Aslan, S., Conze, P.-H., Groza, V., Pham, D. D., Chatterjee, S., Ernst, P., Özkan, S., et al. (2020) · Medical Image Analysis · doi:10.1016/j.media.2020.101950 · Cited by 828 · cited 1× · see citing works
“To evaluate MNAS-Unet, we use the PASCAL VOC 2012 dataset and three medical imaging datasets: PROMISE12 [14], CHAOS [15], and Ultrasound Nerve Segmentation [16], which include MRI, CT, and ultrasound images.”
MedNeXt: Transformer-driven Scaling of ConvNets for Medical Image Segmentation
Roy, S., Koehler, G., Ulrich, C., et al. (2023) · International Conference on Medical Image Computing and Computer-Assisted Intervention · doi:10.48550/arxiv.2303.09975 · Cited by 493 · cited 1×
“The following year, Qing Xu et al. (2023) proposed DCSAU-Net, incorporating split-attention mechanisms to capture fine anatomical structures [4], while the nnUNet team introduced MedNeXt (2023), leveraging large kernel architectures for enhanced global feature extraction [6].”
InfluentialNAS-Unet: Neural Architecture Search for Medical Image Segmentation
Weng, Y., Zhou, T., Li, Y., & Qiu, X. (2019) · IEEE Access · doi:10.1109/access.2019.2908991 · Cited by 487 · cited 1× · see citing works
“NAS-Unet, proposed in 2019, explored differentiable NAS strategies (e.g., DARTS/ProxylessNAS-style methods) on a U-shaped backbone for medical image segmentation [10].”
DCSAU-Net: A deeper and more compact split-attention U-Net for medical image segmentation
Xu, Q., Ma, Z., He, N., & Duan, W. (2023) · Computers in Biology and Medicine · doi:10.1016/j.compbiomed.2023.106626 · Cited by 476 · cited 2× · see citing works
“MNet [5] combined 2D and 3D convolutions for multidimensional information fusion, while DCSAU-Net [4]”
InfluentialChen, C., Dou, Q., Chen, H., Qin, J., & Heng, P. (2020) · IEEE Transactions on Medical Imaging · doi:10.1109/tmi.2020.2972701 · Cited by 442 · cited 1× · see citing works
“To cope with the complexity and diversity of medical images, we design three basic operation sets with special emphasis on stacked DownSC and UpSC cells, aiming to enhance the adaptability of the model to different medical images such as CT and MRI [24].”
Monte Carlo Tree Search: a review of recent modifications and applications
Świechowski, M., Godlewski, K., Sawicki, B., & Mańdziuk, J. (2022) · Artificial Intelligence Review · doi:10.1007/s10462-022-10228-y · Cited by 386 · cited 2× · see citing works
“Swiechowski et al. (2023) provided a comprehensive review of Monte Carlo Tree Search (MCTS) applications in NAS, high-lighting its potential for efficient architecture exploration in resource-constrained environments [9].”
Parallel Monte-Carlo Tree Search
Chaslot, G. M. J.-B., Winands, M., & van den Herik, H. J. (2008) · Computers and Games · doi:10.1007/978-3-540-87608-3_6 · Cited by 310 · see citing works
CARS: Continuous Evolution for Efficient Neural Architecture Search
Yang, Z., Wang, Y., Chen, X., Shi, B., Xu, C., Xu, C., Tian, Q. C., & Xu, C. (2020) · IEEE Computer Society Conference on Computer Vision and Pattern Recognition/Proceedings - IEEE Computer Society Conference on Computer Vision and Pattern Recognition/Proceedings · doi:10.1109/cvpr42600.2020.00190 · Cited by 291 · cited 2× · see citing works
“Yang et al. (2020) proposed CARS, which used continuous evolution as an efficient alternative to differentiable NAS methods [22].”
Lu, H., Li, Y., Uemura, T., Kim, H., & Serikawa, S. (2018) · Future Generation Computer Systems · doi:10.1016/j.future.2018.01.001 · Cited by 290 · see citing works
Lou, A., Guan, S., & Loew, M. H. (2021) · SPIE Medical Imaging: Image Processing · doi:10.1117/12.2582338 · Cited by 212
Underwater vision enhancement technologies: a comprehensive review, challenges, and recent trends
Zhou, J., Yang, T., & Zhang, W. (2023) · Applied Intelligence · doi:10.1007/s10489-022-03767-y · Cited by 149 · cited 1× · see citing works
“The same data augmentation techniques were applied across all models, with additional preprocessing for noisy images [18, 19].”
Review of research on lightweight convolutional neural networks
Zhou, Y., Chen, S., Wang, Y., & Huan, W. (2020) · IEEE Information Technology and Mechatronics Engineering Conference (ITOEC) · doi:10.1109/itoec49072.2020.9141847 · Cited by 133 · cited 1× · see citing works
“This paper proposed an efficient cell search strategy [22] for medical image segmentation and built a lightweight model with fewer parameters based on this strategy [23].”
Neural Architecture Search Using Deep Neural Networks and Monte Carlo Tree Search
Wang, L., Zhao, Y., Jinnai, Y., Tian, Y., & Fonseca, R. (2020) · Proceedings of the AAAI Conference on Artificial Intelligence · doi:10.1609/aaai.v34i06.6554 · Cited by 124 · cited 3× · see citing works
“In this approach, MCTS evaluates the most promising actions at each state, while the MNAS-Unet architecture learns the performance characteristics of sampled architectures and generalizes this knowledge to unexplored architectural configurations [25].”
Rethinking PASCAL-VOC and MS-COCO dataset for small object detection
Tong, K., & Wu, Y. (2023) · Journal of Visual Communication and Image Representation · doi:10.1016/j.jvcir.2023.103830 · Cited by 100 · cited 1× · see citing works
“We investigated proxy cell architectures using the more intricate PASCAL VOC 2012 dataset [20] and then adapted these architectures for medical image datasets.”
Self-aware and Cross-sample Prototypical Learning for Semi-supervised Medical Image Segmentation
Zhang, Z., Ran, R., Tian, C.-N., et al. (2023) · International Conference on Medical Image Computing and Computer-Assisted Intervention · doi:10.48550/arxiv.2305.16214 · Cited by 76 · cited 1×
“To evaluate MNAS-Unet, we use the PASCAL VOC 2012 dataset and three medical imaging datasets: PROMISE12 [14], CHAOS [15], and Ultrasound Nerve Segmentation [16], which include MRI, CT, and ultrasound images.”
Self-aware and Cross-Sample Prototypical Learning for Semi-supervised Medical Image Segmentation
Zhang, Z., Ran, R., Tian, C., Zhou, H., Li, X., Yang, F., & Jiao, Z. (2023) · Lecture notes in computer science · doi:10.1007/978-3-031-43895-0_18 · Cited by 68 · see citing works
MNet: Rethinking 2D/3D Networks for Anisotropic Medical Image Segmentation
Dong, Z., He, Y.-T., Qi, X.-M., et al. (2022) · International Joint Conference on Artificial Intelligence · doi:10.48550/arxiv.2205.04846 · Cited by 66 · cited 2×
“MNet [5] combined 2D and 3D convolutions for multidimensional information fusion, while DCSAU-Net [4]”
Image annotation and curation in radiology: an overview for machine learning practitioners
Galbusera, F., & Cina, A. (2024) · European Radiology Experimental · doi:10.1186/s41747-023-00408-y · Cited by 62 · cited 1× · see citing works
“However, manual segmentation of medical images is labor-intensive, time-consuming, and subject to inter-observer variability, making it impractical for large-scale clinical applications [1, 2].”
MNet: Rethinking 2D/3D Networks for Anisotropic Medical Image Segmentation
Dong, Z., He, Y., Qi, X., et al. (2022) · International Joint Conference on Artificial Intelligence · doi:10.24963/ijcai.2022/122 · Cited by 60 · see citing works
CMM-Net: Contextual multi-scale multi-level network for efficient biomedical image segmentation
Al‐masni, M. A., & Kim, D. (2021) · Scientific Reports · doi:10.1038/s41598-021-89686-3 · Cited by 58 · see citing works
Wang, J., Shu, Y.-C., Lin, C.-Y., Wu, W., Chen, L.-R., Lo, Y., Chiu, H.-C., Özçakar, L., & Chang, K. C. (2023) · Artificial Intelligence in Medicine · doi:10.1016/j.artmed.2023.102496 · Cited by 43 · see citing works
Genetic Neural Architecture Search for automatic assessment of human sperm images
Miahi, E., Mirroshandel, S. A., & Nasr, A. (2022) · Expert Systems with Applications · doi:10.1016/j.eswa.2021.115937 · Cited by 39 · cited 2× · see citing works
“Miahi et al. (2022) explored genetic algorithms in neural architecture search, demonstrating their effectiveness in discovering efficient architectures for medical imaging tasks [31].”
Platon: Pruning large transformer models with upper confidence bound of weight importance
Zhang, Q., Zuo, S., Liang, C., Bukharin, A., He, P., Chen, W., & Zhao, T. (2022) · arXiv · doi:10.48550/arxiv.2206.12562 · Cited by 25 · see citing works
Beyond games: a systematic review of neural Monte Carlo tree search applications
Kemmerling, M., Lütticke, D., & Schmitt, R. H. (2024) · Applied Intelligence · doi:10.1007/s10489-023-05240-w · Cited by 19 · cited 1× · see citing works
“This iterative approach makes MCTS particularly effective for optimizing decision-making in complex domains [35], such as neural architecture search and games, as shown in Fig.”
ResTransUNet: A hybrid CNN-transformer approach for liver and tumor segmentation in CT images
Anwar, A. S., Amin, K., Hadhoud, M. M., & Ibrahim, M. (2025) · Computers in Biology and Medicine · doi:10.1016/j.compbiomed.2025.110048 · Cited by 17 · cited 1× · see citing works
“Most recently, Asmaa Sabet Anwar et al. (2025) proposed ResTransUNet, combining CNN and Transformer architectures for liver and tumor segmentation in CT images [7].”
A Multi-Scale Attention Fusion Network for Retinal Vessel Segmentation
Wang, S.-B., Chen, Y.-Y., & Yi, Z. (2024) · Applied Sciences · doi:10.3390/app14072955 · Cited by 17 · cited 1× · see citing works
“Wang et al. (2024) proposed a multi-scale attention fusion network for retinal vessel segmentation, demonstrating the effectiveness of combining attention mechanisms with multi-scale feature extraction [36].”
El-Awad, Z. (2022) · Entrepreneurship & Regional Development · doi:10.1080/08985626.2022.2143573 · Cited by 12 · cited 1× · see citing works
“Initially, exploration is emphasized to gather sufficient information for making confident decisions in later stages [32].”
Revisiting Training-free NAS Metrics: An Efficient Training-based Method
Yang, T., Yang, L., Jin, X., & Chen, C. (2023) · 2023 IEEE/CVF Winter Conference on Applications of Computer Vision (WACV) · doi:10.1109/wacv56688.2023.00473 · Cited by 9 · cited 2× · see citing works
“Neural Architecture Search (NAS) [21] significantly reduces the manual cost of designing network structures and greatly expands the possibilities for network architecture design.”
A UCB-Based Tree Search Approach to Joint Verification-Correction Strategy for Large-Scale Systems
Xu, P., Deng, X., & Salado, A. (2022) · IEEE Transactions on Systems, Man & Cybernetics. Systems · doi:10.1109/tsmc.2023.3270446 · Cited by 8 · cited 1×
“At each decision point, actions are selected using the Upper Confidence Bound 1 (UCB1) algorithm [29]: Here, 𝑛 ( 𝜎 ) represents the total visitation count for state 𝜎 , calculated as 𝑛 ( 𝜎 ) = ∑ 𝛼 ∈ 𝑛 ( 𝜎, 𝛼 ) , and 𝑐 is a constant that balances exploration and exploitation.”
Sun, M. W., Troxell, D., & Tibshirani, R. (2023) · Scientific Reports · doi:10.1038/s41598-023-43407-0 · Cited by 7 · cited 1× · see citing works
“However, manual segmentation of medical images is labor-intensive, time-consuming, and subject to inter-observer variability, making it impractical for large-scale clinical applications [1, 2].”
Deep Learning for Solving Loading, Packing, Routing, and Scheduling Problems
Elkefi, S., & Layeb, S. B. (2023) · Handbook of Formal Optimization · doi:10.1007/978-981-19-8851-6_16-1 · Cited by 3 · see citing works
A novel direct measure of exploration and exploitation based on attraction basins
Jerebic, J., Mernik, M., Liu, S.-H., et al. (2021) · Expert Systems with Applications · doi:10.1016/j.eswa.2020.114353
Flagged first, then by citations
Research works printed but not linked: 1
- [28][28] Yang, L., Hu, Y., Lu, S., Sun, Z., Mei, J., Zeng, Y., Shi, Z., Han, Y., Li, X., DDSAS: Dynamic and Differentiable Space-Architecture Search , Asian Conference on Machine Learning (ACML), PMLR, 284-299, 2021. -
- Authors
- Meng, L., Nie, F., Zhang, Y., & Han, C.
- Corresponding author
- Meng, Liping
- Venue
- arXiv
- Publisher
- arXiv
- Published
- 2026
- ISSN
- 1556-5068 (ISSN-L)
- arXiv
- 2602.22361
- OpenAlex
- W7131889852
- Semantic Scholar
- 43884361da2cdbde71d27abf8e069c344131ed33
- Cited by
- 0
- References
- 36
- Type
- posted content
- Open access
- green
- doi.org
- Locations
- SSRN Electronic Journal (acceptedVersion)
- arXiv (Cornell University) (submittedVersion)
- arXiv (Cornell University)
- Topics
- Physical Sciences › Computer Science › Computer Vision and Pattern Recognition › Medical Image Segmentation Techniques
- Physical Sciences › Computer Science › Computer Vision and Pattern Recognition › Advanced Neural Network Applications
- Physical Sciences › Computer Science › Artificial Intelligence › Domain Adaptation and Few-Shot Learning
- Keywords
- medical image segmentation, neural architecture search, Monte Carlo tree search, lightweight models, GPU memory consumption, chaos
About these checks
Registry, retraction, Cites and statistics results are computed, not judged, with one exception: a printed reference that no index lists is read once by a model before it counts as a citation error. Claims are extracted by a model and kept only when their quote appears verbatim in the text; each cited source is then checked, with a model’s help, for whether it says what the claim relies on. Those verdicts can be wrong — read the claim’s quote and the cited source before relying on one.
Claims read by deepseek-v4.1-flash under definition version 3 (extract-v3) on 2026-10-22.
Classified by jev-1.13-20260917 (gold-guideline-v0.2) on 2026-10-22.
Cited sources are checked by gpt-6-sol.
Statistics and arithmetic are recomputed, not judged.
Confidence bands are shown only for fields calibrated on our annotated test set. None is calibrated yet, so no band is shown.
What counts as a claim ›How we measure confidence ›How we check this ›
Are you an author, or do you see a problem with this page? Request a review