用生成AI+强化学习实现可复现的自动心脏超声扫描
End-to-End Framework Integrating Generative AI and Deep Reinforcement Learning for Autonomous Ultrasound Scanning
- 用GAN与VAE融合生成逼真超声图像,模拟真实扫描环境
- 强化学习在仿真环境中训练出准确、稳定的自主扫描策略
- 公开真实超声数据集,支持复现与拓展到其他器官
心脏超声是心血管诊断中最常用的工具之一,但其效果受限于操作者依赖、时间压力和人为误差。专业人员短缺,尤其在偏远地区,进一步限制了获取机会。这凸显了自动化解决方案的必要性,以确保无论操作者技能或地理位置如何,都能获得一致且可及的心脏成像。人工智能,特别是深度强化学习(DRL),在实现自主决策方面引起关注。然而,现有的基于DRL的心脏超声扫描方法缺乏可复现性,依赖专有数据,并使用简化模型。为填补这些空白,我们提出首个集成生成AI与DRL的端到端框架,实现可复现的自主心脏超声扫描。该框架包含两部分:(i) 一个结合生成对抗网络(GAN)与变分自编码器(VAE)的条件生成模拟器,用于生成动作条件下的真实感超声图像;(ii) 一个利用该模拟器学习自主、精准扫描策略的DRL模块。所提框架通过专家验证的模型实现图像类型分类与质量评估,支持条件生成真实超声图像,并建立可扩展至其他器官的可复现基础。为确保可复现性,发布了一个公开可用的真实心脏超声扫描数据集。通过多项实验验证:对VAE-GAN与现有GAN变体进行基准测试,采用定性和定量方法评估性能;对基于DRL的扫描系统在不同配置下评估有效性。
原文摘要 · Abstract (English)
Cardiac ultrasound (US) is among the most widely used diagnostic tools in cardiology for assessing heart health, but its effectiveness is limited by operator dependence, time constraints, and human error. The shortage of trained professionals, especially in remote areas, further restricts access. These issues underscore the need for automated solutions that can ensure consistent, and accessible cardiac imaging regardless of operator skill or location. Recent progress in artificial intelligence (AI), especially in deep reinforcement learning (DRL), has gained attention for enabling autonomous decision-making. However, existing DRL-based approaches to cardiac US scanning lack reproducibility, rely on proprietary data, and use simplified models. Motivated by these gaps, we present the first end-to-end framework that integrates generative AI and DRL to enable autonomous and reproducible cardiac US scanning. The framework comprises two components: (i) a conditional generative simulator combining Generative Adversarial Networks (GANs) with Variational Autoencoders (VAEs), that models the cardiac US environment producing realistic action-conditioned images; and (ii) a DRL module that leverages this simulator to learn autonomous, accurate scanning policies. The proposed framework delivers AI-driven guidance through expert-validated models that classify image type and assess quality, supports conditional generation of realistic US images, and establishes a reproducible foundation extendable to other organs. To ensure reproducibility, a publicly available dataset of real cardiac US scans is released. The solution is validated through several experiments. The VAE-GAN is benchmarked against existing GAN variants, with performance assessed using qualitative and quantitative approaches, while the DRL-based scanning system is evaluated under varying configurations to demonstrate effectiveness.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。