arXiv:2511.18083cs.CV2025-11

用简单特征+轻量模型实现高精度疟疾分类,适合资源受限场景

Less Is More: An Explainable AI Framework for Lightweight Malaria Classification

  • 基于细胞像素数和孔洞数提取特征,用逻辑回归与随机森林构建轻量模型
  • 单模型准确率达94.8%,集成模型提升至97.15%,推理仅需2.3毫秒
  • 模型仅1.2KB,比深度学习快30倍,透明可解释,适合基层医疗部署

深度学习模型虽性能优越但计算量大且缺乏可解释性,常用于医学图像分类。本研究探讨复杂神经网络对疟疾二分类任务是否必要。提出可解释、可复现的轻量级特征工程流程EMFE,专为细胞形态设计,可在仅含CPU的设备上达到深度学习水平性能,具备实际部署潜力。使用NIH疟疾细胞图像数据集,从每张图像中提取非背景像素数与细胞内孔洞数两个特征,对比逻辑回归、随机森林与ResNet18、DenseNet121、MobileNetV2、EfficientNet在准确率、模型大小与CPU推理时间上的表现。通过融合逻辑回归与随机森林构建集成模型,进一步提升性能。结果表明,单变量逻辑回归模型测试准确率达94.80%,文件大小仅1.2 kB,推理延迟低至2.3毫秒;两阶段集成模型准确率提升至97.15%。相比之下,深度学习模型占用13.6–44.7 MB存储,推理时间达68毫秒以上。结论显示,基于紧凑特征工程的方法可在保证临床可用性的同时,显著提升透明度、可复现性、速度与部署可行性。该流程表明,简单可解释特征结合轻量模型,是计算资源有限环境下的实用诊断方案。

原文摘要 · Abstract (English)

Background and Objective: Deep learning models have high computational needs and lack interpretability but are often the first choice for medical image classification tasks. This study addresses whether complex neural networks are essential for the simple binary classification task of malaria. We introduce the Extracted Morphological Feature Engineered (EMFE) pipeline, a transparent, reproducible, and low compute machine learning approach tailored explicitly for simple cell morphology, designed to achieve deep learning performance levels on a simple CPU only setup with the practical aim of real world deployment. Methods: The study used the NIH Malaria Cell Images dataset, with two features extracted from each cell image: the number of non background pixels and the number of holes within the cell. Logistic Regression and Random Forest were compared against ResNet18, DenseNet121, MobileNetV2, and EfficientNet across accuracy, model size, and CPU inference time. An ensemble model was created by combining Logistic Regression and Random Forests to achieve higher accuracy while retaining efficiency. Results: The single variable Logistic Regression model achieved a test accuracy of 94.80 percent with a file size of 1.2 kB and negligible inference latency (2.3 ms). The two stage ensemble improved accuracy to 97.15 percent. In contrast, the deep learning methods require 13.6 MB to 44.7 MB of storage and show significantly higher inference times (68 ms). Conclusion: This study shows that a compact feature engineering approach can produce clinically meaningful classification performance while offering gains in transparency, reproducibility, speed, and deployment feasibility. The proposed pipeline demonstrates that simple interpretable features paired with lightweight models can serve as a practical diagnostic solution for environments with limited computational resources.

医疗影像轻量模型可解释AI疟疾检测

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。