arXiv:2507.00743eess.IVcs.CV2025-07被引 2

用可调小波单元提升OCT图像分类精度,区分视网膜手术类型。

Tunable Wavelet Unit based Convolutional Neural Network in Optical Coherence Tomography Analysis Enhancement for Classifying Type of Epiretinal Membrane Surgery

  • 在ResNet18中嵌入可调小波单元,自动优化滤波参数。
  • 小波预处理后模型准确率达78%,优于原始图像的66%。
  • 首次将可调小波用于区分视网膜膜切除术类型,适合眼科AI研究者。

本研究开发了一种基于深度学习的方法,用于分类视网膜前膜(ERM)切除手术类型,包括内界膜(ILM)一并切除或仅切除ERM。模型基于ResNet18卷积神经网络架构,以术后光学相干断层扫描(OCT)中心切片为输入。在原始扫描与能量裁剪及小波去噪预处理后的扫描上分别评估,预处理后准确率达到72%,高于原始扫描的66%。为进一步提升性能,引入两种关键改进的可调小波单元:基于正交格栅的小波单元(OrthLatt-UwU)和基于完美重构松弛的小波单元(PR-Relax-UwU)。这些单元可训练过程中自动调整滤波系数,并嵌入下采样、步长为2的卷积及池化层,显著增强对两类手术类型的区分能力,其中OrthLatt-UwU使准确率提升至76%,PR-Relax-UwU达78%。对比显示,该AI模型性能超过人工阅片者(仅50%准确率)。据我们所知,这是首个将可调小波应用于不同类型ERM切除手术分类的工作,展示了基于CNN模型在临床决策支持中的潜力。

原文摘要 · Abstract (English)

In this study, we developed deep learning-based method to classify the type of surgery performed for epiretinal membrane (ERM) removal, either internal limiting membrane (ILM) removal or ERM-alone removal. Our model, based on the ResNet18 convolutional neural network (CNN) architecture, utilizes postoperative optical coherence tomography (OCT) center scans as inputs. We evaluated the model using both original scans and scans preprocessed with energy crop and wavelet denoising, achieving 72% accuracy on preprocessed inputs, outperforming the 66% accuracy achieved on original scans. To further improve accuracy, we integrated tunable wavelet units with two key adaptations: Orthogonal Lattice-based Wavelet Units (OrthLatt-UwU) and Perfect Reconstruction Relaxation-based Wavelet Units (PR-Relax-UwU). These units allowed the model to automatically adjust filter coefficients during training and were incorporated into downsampling, stride-two convolution, and pooling layers, enhancing its ability to distinguish between ERM-ILM removal and ERM-alone removal, with OrthLattUwU boosting accuracy to 76% and PR-Relax-UwU increasing performance to 78%. Performance comparisons showed that our AI model outperformed a trained human grader, who achieved only 50% accuracy in classifying the removal surgery types from postoperative OCT scans. These findings highlight the potential of CNN based models to improve clinical decision-making by providing more accurate and reliable classifications. To the best of our knowledge, this is the first work to employ tunable wavelets for classifying different types of ERM removal surgery.

医学影像小波分析手术分类深度学习

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