arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2510.01547cs.CVcs.LG

有限训练数据下的口腔癌鲁棒分类

Robust Classification of Oral Cancer with Limited Training Data

The University of Texas at Dallas(德克萨斯大学达拉斯分校);Quality of Life Technology Laboratory(生活质量技术实验室)

Akshay Bhagwan Sonawane, Lena D. Swamikannan, Lakshman Tamil

AI总结:

针对训练数据有限的口腔癌分类场景,提出结合CNN与贝叶斯深度学习、采用变分推断的混合模型,在小样本下提升了模型泛化性与可靠性,助力口腔癌早期诊断。

摘要:

口腔癌是全球最常见的癌症之一,在医疗资源匮乏的地区死亡率尤其高。早期诊断对降低死亡率至关重要,但由于口腔健康项目不足、基础设施不完善以及医护人员短缺,相关挑战依然存在。传统深度学习模型虽有前景,但通常依赖点估计,会导致预测过度自信,可靠性下降。关键问题在于,这类模型需要大量数据集来缓解过拟合并保证泛化能力,这在训练数据有限的场景下是不现实的需求。为解决这些问题,我们提出一种混合模型,将卷积神经网络(CNN)与贝叶斯深度学习结合,用于小训练集下的口腔癌分类。该方法采用变分推断,通过不确定性量化提升可靠性。模型在智能手机拍摄的彩色照片图像上训练,并在三个不同的测试数据集上进行评估。在分布与训练数据相似的测试数据集上,所提方法达到了94%的准确率,与传统CNN的性能相当。值得注意的是,对于真实场景的拍摄图像数据,尽管存在与训练数据集不同的限制和差异,所提模型展现出更优的泛化能力:即使使用更小的数据集,其在多样化数据集上的准确率达到88%,而传统CNN仅为72.94%。置信度分析显示,该模型对正确分类的样本表现出低不确定性(高置信度),对错分样本则表现出高不确定性(低置信度)。这些结果凸显了在数据稀缺环境中,贝叶斯推断通过提升模型可靠性和泛化能力,在增强口腔癌早期诊断方面的有效性。

英文摘要:

Oral cancer ranks among the most prevalent cancers globally, with a particularly high mortality rate in regions lacking adequate healthcare access. Early diagnosis is crucial for reducing mortality; however, challenges persist due to limited oral health programs, inadequate infrastructure, and a shortage of healthcare practitioners. Conventional deep learning models, while promising, often rely on point estimates, leading to overconfidence and reduced reliability. Critically, these models require large datasets to mitigate overfitting and ensure generalizability, an unrealistic demand in settings with limited training data. To address these issues, we propose a hybrid model that combines a convolutional neural network (CNN) with Bayesian deep learning for oral cancer classification using small training sets. This approach employs variational inference to enhance reliability through uncertainty quantification. The model was trained on photographic color images captured by smartphones and evaluated on three distinct test datasets. The proposed method achieved 94% accuracy on a test dataset with a distribution similar to that of the training data, comparable to traditional CNN performance. Notably, for real-world photographic image data, despite limitations and variations differing from the training dataset, the proposed model demonstrated superior generalizability, achieving 88% accuracy on diverse datasets compared to 72.94% for traditional CNNs, even with a smaller dataset. Confidence analysis revealed that the model exhibits low uncertainty (high confidence) for correctly classified samples and high uncertainty (low confidence) for misclassified samples. These results underscore the effectiveness of Bayesian inference in data-scarce environments in enhancing early oral cancer diagnosis by improving model reliability and generalizability.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑