用泰勒展开的样条网络实现轻量级医疗影像分类,精准且可解释。
Taylor-Series Expanded Kolmogorov-Arnold Network for Medical Imaging Classification
- 基于样条与泰勒展开的KAN模型,直接处理原始医学图像
- 仅用30%数据达86%准确率,参数量仅2872,远低于传统CNN
- 适用于资源有限的临床场景,且支持可视化解释
在资源受限的临床环境中,高效且可解释的医学影像分类仍是计算机辅助诊断的挑战。本文提出基于样条的柯尔莫哥洛夫-阿诺德网络(KAN),包括SBTAYLOR-KAN(B样条+泰勒展开)、SBRBF-KAN(B样条+径向基函数)和SBWAVELET-KAN(B样条+莫莱特小波),利用样条函数逼近捕捉局部与全局非线性特征。模型在脑部MRI、胸部X光、结核X光及皮肤病变图像上无需预处理即可直接学习,跨数据集验证和数据缩减分析显示其具备强泛化能力与稳定性。SBTAYLOR-KAN最高达98.93%准确率,在三个数据集上仅用30%训练数据仍保持超86%准确率;在皮肤癌数据集(存在类别不平衡)中,无论是否平衡,均优于其他模型,达68.22%准确率。相比传统CNN(如ResNet50,2418万参数),该模型仅需2872个可训练参数,显著降低计算负担。通过梯度加权类激活映射(Grad-CAM)实现可视化解释,突出关键病灶区域。该框架为医疗影像分类提供了轻量、可解释、高泛化的解决方案,适用于数据稀缺的临床人工智能场景。
原文摘要 · Abstract (English)
Effective and interpretable classification of medical images is a challenge in computer-aided diagnosis, especially in resource-limited clinical settings. This study introduces spline-based Kolmogorov-Arnold Networks (KANs) for accurate medical image classification with limited, diverse datasets. The models include SBTAYLOR-KAN, integrating B-splines with Taylor series; SBRBF-KAN, combining B-splines with Radial Basis Functions; and SBWAVELET-KAN, embedding B-splines in Morlet wavelet transforms. These approaches leverage spline-based function approximation to capture both local and global nonlinearities. The models were evaluated on brain MRI, chest X-rays, tuberculosis X-rays, and skin lesion images without preprocessing, demonstrating the ability to learn directly from raw data. Extensive experiments, including cross-dataset validation and data reduction analysis, showed strong generalization and stability. SBTAYLOR-KAN achieved up to 98.93% accuracy, with a balanced F1-score, maintaining over 86% accuracy using only 30% of the training data across three datasets. Despite class imbalance in the skin cancer dataset, experiments on both imbalanced and balanced versions showed SBTAYLOR-KAN outperforming other models, achieving 68.22% accuracy. Unlike traditional CNNs, which require millions of parameters (e.g., ResNet50 with 24.18M), SBTAYLOR-KAN achieves comparable performance with just 2,872 trainable parameters, making it more suitable for constrained medical environments. Gradient-weighted Class Activation Mapping (Grad-CAM) was used for interpretability, highlighting relevant regions in medical images. This framework provides a lightweight, interpretable, and generalizable solution for medical image classification, addressing the challenges of limited datasets and data-scarce scenarios in clinical AI applications.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。