用轻量NAS算法自动搜寻适合早筛脑瘫的神经网络,提升诊断准确率。
Lightweight Neural Architecture Search for Cerebral Palsy Detection
- 基于强化学习的轻量级神经架构搜索,自动优化模型结构与超参数。
- 在真实脑瘫数据集上表现优于现有方法,且无需复杂集成模型。
- 模型轻量高效,适合资源有限地区部署,助力基层医疗早筛。
脑瘫(CP)是一种在婴幼儿期显现的神经系统疾病,长期影响运动协调能力,是儿童残疾的主要原因,早期检测对及时治疗至关重要。目前依赖专家进行一般运动评估(GMA),但该方法在发展中国家难以普及。传统机器学习预测性能有限,现有专家方案多为特定数据集定制,泛化能力差。为此,我们提出一种基于强化学习更新机制的神经架构搜索(NAS)算法,可高效优化最佳网络结构与超参数组合,发现最适合脑瘫检测的神经网络配置。该方法在真实脑瘫数据集上的表现优于领域内其他依赖大型集成模型的方法。由于其资源消耗低、计算效率高,特别适用于医疗资源匮乏的农村或发展中国家。所获模型具有轻量化架构和快速推理时间,可在算力有限设备上部署,减少对昂贵基础设施的需求,可嵌入临床流程,为早期脑瘫诊断提供及时准确支持。
原文摘要 · Abstract (English)
The neurological condition known as cerebral palsy (CP) first manifests in infancy or early childhood and has a lifelong impact on motor coordination and body movement. CP is one of the leading causes of childhood disabilities, and early detection is crucial for providing appropriate treatment. However, such detection relies on assessments by human experts trained in methods like general movement assessment (GMA). These are not widely accessible, especially in developing countries. Conventional machine learning approaches offer limited predictive performance on CP detection tasks, and the approaches developed by the few available domain experts are generally dataset-specific, restricting their applicability beyond the context for which these were created. To address these challenges, we propose a neural architecture search (NAS) algorithm applying a reinforcement learning update scheme capable of efficiently optimizing for the best architectural and hyperparameter combination to discover the most suitable neural network configuration for detecting CP. Our method performs better on a real-world CP dataset than other approaches in the field, which rely on large ensembles. As our approach is less resource-demanding and performs better, it is particularly suitable for implementation in resource-constrained settings, including rural or developing areas with limited access to medical experts and the required diagnostic tools. The resulting model's lightweight architecture and efficient computation time allow for deployment on devices with limited processing power, reducing the need for expensive infrastructure, and can, therefore, be integrated into clinical workflows to provide timely and accurate support for early CP diagnosis.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。