用记忆网络增强LLM,让X光报告更准地描述疾病
Activating Associative Disease-Aware Vision Token Memory for LLM-Based X-ray Report Generation
- 引入视觉霍普菲尔德网络,关联疾病区域与报告记忆
- 在3个数据集上达到当前最优,显著提升疾病描述准确率
- 适合医学影像生成、临床辅助诊断方向研究者
近年来,基于大语言模型的X光图像报告生成取得了显著进展,但这些模型未能充分挖掘图像中与疾病相关的有效视觉信息,导致报告虽语言流畅,却对关键疾病的描述不足。本文提出一种新型关联记忆增强型X光报告生成模型,有效模拟专业医生撰写报告的过程。该模型同时挖掘全局与局部视觉信息,并关联历史报告信息以更好完成当前报告。具体而言,给定一张X光图像,先通过分类模型及其激活图定位高度相关于疾病的视觉区域,并学习疾病查询令牌;随后,利用视觉霍普菲尔德网络建立疾病相关令牌的记忆关联,报告霍普菲尔德网络则用于检索报告记忆信息。这一过程基于大语言模型生成高质量报告,在多个基准数据集(包括IU X-ray、MIMIC-CXR和Chexpert Plus)上取得领先性能。代码已公开于https://github.com/Event-AHU/Medical_Image_Analysis。
原文摘要 · Abstract (English)
X-ray image based medical report generation achieves significant progress in recent years with the help of the large language model, however, these models have not fully exploited the effective information in visual image regions, resulting in reports that are linguistically sound but insufficient in describing key diseases. In this paper, we propose a novel associative memory-enhanced X-ray report generation model that effectively mimics the process of professional doctors writing medical reports. It considers both the mining of global and local visual information and associates historical report information to better complete the writing of the current report. Specifically, given an X-ray image, we first utilize a classification model along with its activation maps to accomplish the mining of visual regions highly associated with diseases and the learning of disease query tokens. Then, we employ a visual Hopfield network to establish memory associations for disease-related tokens, and a report Hopfield network to retrieve report memory information. This process facilitates the generation of high-quality reports based on a large language model and achieves state-of-the-art performance on multiple benchmark datasets, including the IU X-ray, MIMIC-CXR, and Chexpert Plus. The source code of this work is released on \url{https://github.com/Event-AHU/Medical_Image_Analysis}.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。