arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2307.03273cs.CVeess.IV

ADASSM: 基于图像的统计形状模型中的对抗数据增强

ADASSM: Adversarial Data Augmentation in Statistical Shape Models From Images

Scientific Computing and Imaging Institute, University of Utah(犹他大学科学计算与成像研究所);Kahlert School of Computing, University Of Utah(犹他大学卡勒特计算学院)

Mokshagna Sai Teja Karanam, Tushar Kataria, Krithika Iyer, Shireen Elhabian

AI总结:

针对医学数据有限导致Image-to-SSM网络易过拟合、现有增强方法仅关注形状而存在纹理偏差的问题,提出ADASSM对抗纹理增强框架,通过生成噪声样本引导模型关注几何结构,提升了模型精度。

摘要:

统计形状模型(SSM)已被公认为是识别目标人群中解剖结构形态变异的优良工具。形状模型在给定队列的所有样本中使用一致的形状表示,这有助于对比形状并识别可用于检测病理、辅助制定治疗方案的变异。在医学成像领域,从CT/MRI扫描中计算这些形状表示需要耗时的预处理操作,包括但不限于解剖结构分割标注、配准以及纹理去噪。深度学习模型在直接从体图像中学习形状表示方面展现出卓越能力,由此催生了高效且高性价比的Image-to-SSM网络。尽管如此,这些模型对数据需求量大,且由于医学数据获取有限,深度学习模型容易出现过拟合。采用基于核密度估计(KDE)方法生成形状增强样本的离线数据增强技术,已成功助力Image-to-SSM网络达到与传统SSM方法相当的精度。然而,这些增强方法仅聚焦于形状增强,而深度学习模型存在基于图像的纹理偏差,会导致模型性能未达最优。本文提出了一种面向Image-to-SSM框架的新型实时数据增强策略,该策略利用数据依赖的噪声生成即纹理增强实现。所提出的框架被训练为Image-to-SSM网络的对抗方,可生成多样且具有挑战性的噪声样本。我们的方法通过促使模型关注底层几何结构而非仅依赖像素值,实现了精度的提升。

英文摘要:

Statistical shape models (SSM) have been well-established as an excellent tool for identifying variations in the morphology of anatomy across the underlying population. Shape models use consistent shape representation across all the samples in a given cohort, which helps to compare shapes and identify the variations that can detect pathologies and help in formulating treatment plans. In medical imaging, computing these shape representations from CT/MRI scans requires time-intensive preprocessing operations, including but not limited to anatomy segmentation annotations, registration, and texture denoising. Deep learning models have demonstrated exceptional capabilities in learning shape representations directly from volumetric images, giving rise to highly effective and efficient Image-to-SSM networks. Nevertheless, these models are data-hungry and due to the limited availability of medical data, deep learning models tend to overfit. Offline data augmentation techniques, that use kernel density estimation based (KDE) methods for generating shape-augmented samples, have successfully aided Image-to-SSM networks in achieving comparable accuracy to traditional SSM methods. However, these augmentation methods focus on shape augmentation, whereas deep learning models exhibit image-based texture bias resulting in sub-optimal models. This paper introduces a novel strategy for on-the-fly data augmentation for the Image-to-SSM framework by leveraging data-dependent noise generation or texture augmentation. The proposed framework is trained as an adversary to the Image-to-SSM network, augmenting diverse and challenging noisy samples. Our approach achieves improved accuracy by encouraging the model to focus on the underlying geometry rather than relying solely on pixel values.

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

↑