可控生成增强提升医疗序列分类,解决合成数据质量与语义一致性难题
Ctrl-GenAug: Controllable Generative Augmentation for Medical Sequence Classification
- 通过多模态条件引导生成诊断相关序列样本
- 在3个医疗数据集上提升模型性能,尤其在罕见人群和域外场景下
- 引入噪声过滤机制,抑制错误合成样本,保障数据可靠性
医疗领域受限于大规模数据集稀缺和人工标注耗时,深度模型性能受限。基于扩散的生成增强方法虽有潜力,但现有工作在复杂视频/3D序列生成中缺乏语义与时间上的可控性,且忽略噪声合成样本的质量控制,导致合成数据库不可靠,严重制约下游任务表现。本文提出Ctrl-GenAug,一种通用生成增强框架,可实现高度语义与时间定制的序列合成,并抑制错误样本。首先设计多模态条件引导的序列生成器,可控生成促进诊断的样本;集成序列增强模块以提升生成样本的时间与立体一致性;进而提出噪声合成数据过滤器,在语义与序列层面抑制不可靠案例。在3个医疗数据集上,使用11种网络、3种训练范式进行广泛实验,全面验证了该方法在低频高危人群及域外条件下的有效性与通用性。
原文摘要 · Abstract (English)
In the medical field, the limited availability of large-scale datasets and labor-intensive annotation processes hinder the performance of deep models. Diffusion-based generative augmentation approaches present a promising solution to this issue, having been proven effective in advancing downstream medical recognition tasks. Nevertheless, existing works lack sufficient semantic and sequential steerability for challenging video/3D sequence generation, and neglect quality control of noisy synthesized samples, resulting in unreliable synthetic databases and severely limiting the performance of downstream tasks. In this work, we present Ctrl-GenAug, a novel and general generative augmentation framework that enables highly semantic- and sequential-customized sequence synthesis and suppresses incorrectly synthesized samples, to aid medical sequence classification. Specifically, we first design a multimodal conditions-guided sequence generator for controllably synthesizing diagnosis-promotive samples. A sequential augmentation module is integrated to enhance the temporal/stereoscopic coherence of generated samples. Then, we propose a noisy synthetic data filter to suppress unreliable cases at semantic and sequential levels. Extensive experiments on 3 medical datasets, using 11 networks trained on 3 paradigms, comprehensively analyze the effectiveness and generality of Ctrl-GenAug, particularly in underrepresented high-risk populations and out-domain conditions.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。