摘要:开发一种基于深度学习的唐氏综合症智能诊断辅助系统,通过面部图像特征自动识别唐氏综合症患者,提高早期筛查效率和准确性。
项目概览
项目简介
本研究采用渐进式训练策略,构建了基于EfficientNet-B0和DenseNet121的双模型深度学习框架。数据集包含2,986张300×300像素的面部图像(唐氏综合症样本1,487张,健康对照样本1,499张),其中1,561张图像包含面部测量特征标注(眼间距、眼口距离)。采用数据增强技术(随机水平翻转、旋转、亮度调整)和类别加权损失函数处理数据不平衡问题。训练过程分为两个阶段:阶段一使用EfficientNet-B0进行20轮快速基线验证;阶段二使用DenseNet121进行30轮深度优化。采用分层抽样将数据集按70:15:15比例划分为训练集、验证集和测试集。
EfficientNet-B0模型在测试集上达到94.47%的准确率和0.9873的AUC值。DenseNet121模型表现更优,测试准确率达96.60%,AUC值为0.9944,接近完美分类性能。针对健康对照组,DenseNet121的精确率和召回率均为94.20%;针对唐氏综合症组,精确率和召回率均达97.59%。训练过程中,DenseNet121在第24轮达到最佳验证准确率98.72%,最终训练准确率收敛至99.91%。
本研究成功开发了高精度的唐氏综合症智能诊断辅助系统,验证了深度学习在医学影像辅助诊断领域的有效性。系统集成了图形用户界面,实现了临床易用性。DenseNet121模型表现出色的分类性能和临床应用潜力,可作为唐氏综合症早期筛查的有效工具。未来研究方向包括扩大数据集规模、融合多模态临床特征、以及开发移动端应用以提高临床可及性。
系统架构
本系统采用基于深度学习的端到端架构设计,包含数据处理层、模型推理层和用户交互层三个核心模块。数据处理层负责面部图像的预处理,包括尺寸归一化(224×224像素)、数据增强(随机翻转、旋转、亮度调整)和标准化处理;模型推理层实现了渐进式训练策略,先通过EfficientNet-B0(5.3M参数)进行20轮快速基线验证,再使用DenseNet121(8.0M参数)进行30轮深度优化,两个模型均采用ImageNet预训练权重进行迁移学习,使用AdamW优化器和交叉熵损失函数(类别加权2.39:1处理数据不平衡),并通过ReduceLROnPlateau学习率调度器动态调整训练过程;用户交互层采用PyQt5框架构建现代化图形界面,支持实时图像预览、模型选择、置信度可视化和概率分布展示,整个系统在NVIDIA GPU加速下实现毫秒级推理速度,为临床医生提供高效准确的唐氏综合症辅助诊断工具。
图1 系统架构图
技术创新
创新点1:渐进式双模型训练策略
本研究提出了一种渐进式双模型训练策略,通过两阶段优化实现快速验证与深度优化的有机结合。第一阶段采用轻量级EfficientNet-B0(5.3M参数)进行20轮快速基线验证,在保证计算效率的同时快速确定可行性;第二阶段使用深层DenseNet121(8.0M参数)进行30轮深度优化,充分挖掘面部特征的判别能力。该策略不同于传统单一模型训练方法,通过模型复杂度的渐进提升,在训练效率和模型性能之间取得最优平衡,最终DenseNet121在测试集上达到96.60%的准确率和0.9944的AUC值,显著优于单一训练策略。
创新点2:类别加权损失与数据增强融合的不平衡数据处理机制
针对医学数据集普遍存在的类别不平衡问题(本研究中唐氏综合症与健康样本比例2.39:1),本研究设计了类别加权损失函数与多维度数据增强相结合的处理机制。在损失函数层面,采用交叉熵损失配合类别权重(健康:唐氏=2.39:1),使模型在训练过程中更关注少数类样本;在数据增强层面,设计了针对面部图像特点的增强策略,包括随机水平翻转(p=0.5)、旋转(±15°)、颜色抖动(亮度/对比度/饱和度±0.2)和仿射变换(平移±10%),在保持面部语义信息的同时扩充少数类样本多样性。该融合机制有效缓解了不平衡数据对模型性能的负面影响,使健康类别(少数类)的精确率和召回率达到94.20%,唐氏综合症类别(多数类)达到97.59%,两类性能均衡优异。
数据集
本研究使用的数据集包含2,986张300×300像素的面部图像,其中唐氏综合症样本1,487张,健康对照样本1,499张,类别分布基本均衡(1:1.008);数据集配备Excel标注文件(labeling.xlsx)包含1,561条有效标注记录,每条记录包含图像路径、眼间距(eyetoeye)、眼到嘴距离(eyetomouth)和类别标签(0=健康,1=唐氏综合症)四个字段;数据采用分层抽样按70:15:15比例划分为训练集(1,092样本)、验证集(234样本)和测试集(235样本),确保各集合类别分布一致性,为模型训练和评估提供可靠基础。
模型训练
运行train.py
图2 训练历史曲线(Loss和Accuracy)
图3 ROC曲线和AUC值
图4 混淆矩阵热力图
图2 训练历史曲线(Loss和Accuracy)
图3 ROC曲线和AUC值
图4 混淆矩阵热力图
快速开始
执行 pip install -r requirements.txt 安装依赖后,运行 python train.py 开始训练,训练完成后执行 python gui.py 启动图形界面进行唐氏综合症面部图像检测。
环境要求
本系统基于Python 3.8+和PyTorch 2.0+开发,推荐配置为Windows 10/11系统、16GB内存、NVIDIA GPU(支持CUDA加速),必需依赖包括torchvision、pandas、pillow、matplotlib、seaborn、scikit-learn、tqdm、openpyxl和PyQt5。
运行展示
运行gui.py
图8 主界面图
图9 EfficientNet分析结果:唐氏综合症
图10 EfficientNet分析结果:健康
图11 DenseNet分析结果:唐氏综合症
图12 DenseNet分析结果:健康
项目资源
配套文件
包括完整的项目源代码、演示视频、运行截图,开箱即用。
项目信息
作者信息
作者:Bob (张家梁)
项目编号:AI-33-P
原创声明:本项目为原创作品