论文标题
加强医疗报告生成,并以X线性的关注和重复罚款
Reinforced Medical Report Generation with X-Linear Attention and Repetition Penalty
论文作者
论文摘要
为了减少医生的工作量,基于深度学习的自动医疗报告的一代最近吸引了越来越多的研究工作,其中注意力机制和增强学习与经典的编码器核对器架构集成在一起,以增强深层模型的性能。但是,这些最先进的解决方案主要遭受两个缺点:(i)他们的注意机制无法利用高阶特征相互作用,并且(ii)由于使用了基于TF-IDF的奖励功能,这些方法具有生成重复术语的脆弱性。因此,在这项工作中,我们提出了一种增强的医学报告生成解决方案,并具有X线性的关注和重复惩罚机制(REMRG-XR)来克服这些问题。具体而言,X线性注意模块用于探索高级特征交互并实现多模式推理,而重复惩罚用于在模型的训练过程中对重复条款进行惩罚。在两个公共数据集上进行了广泛的实验研究,结果表明,REMRG-XR在所有指标方面都大大优于最先进的基线。
To reduce doctors' workload, deep-learning-based automatic medical report generation has recently attracted more and more research efforts, where attention mechanisms and reinforcement learning are integrated with the classic encoder-decoder architecture to enhance the performance of deep models. However, these state-of-the-art solutions mainly suffer from two shortcomings: (i) their attention mechanisms cannot utilize high-order feature interactions, and (ii) due to the use of TF-IDF-based reward functions, these methods are fragile with generating repeated terms. Therefore, in this work, we propose a reinforced medical report generation solution with x-linear attention and repetition penalty mechanisms (ReMRG-XR) to overcome these problems. Specifically, x-linear attention modules are used to explore high-order feature interactions and achieve multi-modal reasoning, while repetition penalty is used to apply penalties to repeated terms during the model's training process. Extensive experimental studies have been conducted on two public datasets, and the results show that ReMRG-XR greatly outperforms the state-of-the-art baselines in terms of all metrics.