MedRAT: Unpaired Medical Report Generation via Auxiliary Tasks

Elad Hirsch,Gefen Dawidowicz,Ayellet Tal
2024-07-22
Abstract:Medical report generation from X-ray images is a challenging task, particularly in an unpaired setting where paired image-report data is unavailable for training. To address this challenge, we propose a novel model that leverages the available information in two distinct datasets, one comprising reports and the other consisting of images. The core idea of our model revolves around the notion that combining auto-encoding report generation with multi-modal (report-image) alignment can offer a solution. However, the challenge persists regarding how to achieve this alignment when pair correspondence is absent. Our proposed solution involves the use of auxiliary tasks, particularly contrastive learning and classification, to position related images and reports in close proximity to each other. This approach differs from previous methods that rely on pre-processing steps, such as using external information stored in a knowledge graph. Our model, named MedRAT, surpasses previous state-of-the-art methods, demonstrating the feasibility of generating comprehensive medical reports without the need for paired data or external tools.
Computer Vision and Pattern Recognition
What problem does this paper attempt to address?
This paper attempts to solve the problem of automatically generating medical reports in the absence of paired image - report data. Specifically, the paper focuses on how to generate corresponding medical reports from chest X - ray images, and paired image and report data cannot be used during the training process. This is a challenging task because traditional supervised learning methods rely on paired data for training. ### Core Problems of the Paper 1. **Lack of Paired Data**: Due to privacy issues, difficulties in obtaining high - quality data, and the expert knowledge required for medical data analysis and annotation, paired image - report datasets are relatively small and difficult to obtain. 2. **Multi - modal Alignment**: In the absence of a clear image - report correspondence, how to represent the image and the report in a shared space and make them align with each other to generate accurate reports. ### Overview of the Solution To solve the above problems, the authors propose a new model named MedRAT (Medical Report Generation via Auxiliary Tasks). The main innovations of this model include: 1. **Utilizing Auxiliary Tasks**: By introducing auxiliary tasks such as contrastive learning and multi - label classification, relevant images and reports are made closer in the embedding space, while non - relevant ones are made farther. These tasks help the model achieve multi - modal alignment in the absence of paired data. - **Contrastive Learning**: By defining positive sample pairs (images and reports sharing pathological features) and negative sample pairs (those not sharing pathological features), the model can distinguish between semantically similar and dissimilar data points. - **Multi - label Classification**: Predict the pathological label of each sample to compensate for partially matched sample pairs and ensure that the model can correctly classify each example. 2. **Shared Encoder - Decoder Architecture**: This architecture can process text and image data simultaneously during the training process and generate reports using only images during the inference stage. Specifically: - **Global and Local Representations**: The model learns not only global representations (capturing overall information, such as the presence of a pathology), but also local representations (capturing detailed information, such as location, size, and relationships with other organs). - **Self - attention Mechanism**: Aggregate local representations through the self - attention mechanism to generate global representations, thereby better capturing the essence of the data. 3. **Shared Memory Module**: This module records useful feature information and connections, enabling the model to learn from past data and apply it to new inputs. Unlike hand - designed knowledge graphs, this knowledge is automatically learned through end - to - end training. ### Experimental Results The experimental results show that MedRAT outperforms existing state - of - the - art methods on multiple evaluation metrics, especially in natural language generation (NLG) and clinical effectiveness (CE). Specifically: - **Natural Language Generation (NLG)**: MedRAT performs well on metrics such as BLEU, METEOR, ROUGE - L, RadGraph F1, and BERTScore, especially on the MIMIC - CXR dataset. - **Clinical Effectiveness (CE)**: MedRAT also outperforms other methods in terms of precision, recall, and F1 - score, indicating that the reports it generates are more accurate and useful in extracting clinical information (such as pathological features). In conclusion, MedRAT successfully achieves high - quality medical report generation in the absence of paired data by introducing novel auxiliary tasks and a shared encoder - decoder architecture.