针对心脏核磁分割,提出资源感知的神经架构搜索方法。
Resource-Aware Evolutionary Neural Architecture Search for Cardiac MRI Segmentation

- 基于类似UNet的超网,结合心脏特性设计搜索空间。
- 在固定算力下,达93.22%平均DSC与4.73mm HD95。
- 适合需要高效部署的心脏影像分析场景。
心脏磁共振(CMR)分割对心室结构与功能的定量评估至关重要,但低组织对比度、模糊边界及扫描差异使其难以准确分割。本文提出CardiacNAS,一种资源感知的进化神经架构搜索框架,结合类UNet超网与涵盖深度、宽度、卷积核大小、滤波器尺寸、注意力机制、特征融合、激活函数、丢弃率及残差缩放等要素的心脏感知搜索空间。搜索过程显式考虑资源约束,在固定计算预算下联合优化骰子相似系数(DSC)与95百分位豪斯多夫距离(HD95),以模型规模和浮点运算量(FLOPs)为指标。候选架构从超网中实例化,通过代理预算训练,并经交叉、变异与精英选择演化。在ACDC数据集上评估,相较六种先进方法,所获模型达到平均93.22% DSC、4.73 mm HD95,仅需358万参数与14.56 GFLOPs,展现良好精度-效率权衡。分析表明,搜索出的注意力与融合策略,结合残差缩放,显著提升边界保真度与稳定性。CardiacNAS提供了一种可部署、透明报告复杂度与算力预算的系统性解决方案。
原文摘要 · Abstract (English)
Cardiac magnetic resonance (CMR) segmentation underpins quantitative assessment of ventricular structure and function, yet reliable delineation remains difficult due to low tissue contrast, fuzzy boundaries, and inter scan variability. We present CardiacNAS, an evolutionary neural architecture search (NAS) framework that couples a UNet like supernet with a cardiac aware search space spanning depth width, kernel size, filter size, attention, fusion, activation, dropout, and residual scaling. The search is explicitly resource aware, jointly optimizing dice similarity coefficient (DSC) and 95th percentile Hausdorff distance (HD95) versus model size and floating point operations (FLOPs) under fixed compute budgets. Candidate architectures are instantiated from the supernet, trained with proxy budgets, and evolved through crossover, mutation, and elitist selection. We evaluate on the ACDC dataset and compare against six state of the art methods, using qualitative comparisons, learning curve analyses, and design factor correlation studies. The resulting model attains 93.22% average DSC and 4.73 mm HD95 with 3.58M parameters and 14.56 GFLOPs, demonstrating a favorable accuracy efficiency trade off. Analyses indicate that searched attention and fusion choices, together with residual scaling, contribute to improved boundary fidelity and stability. CardiacNAS offers a principled, resource aware approach to deployable CMR segmentation with transparent reporting of architectural complexity and compute budgets.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。