Source-linked AI summary
Efficient Multi-Granularity Knowledge Transfer for Radiology Report Generation
Xubin Zhong, Zheyu Zhang, Wenjian Qin, Ning Wen
TL;DR
Radiology report generation must identify diseases and describe pivotal areas using medical knowledge, but existing methods often overlook multi-granularity knowledge extraction. EMGKT combines global medical vision-language embeddings, fine-grained distillation, and disease-diagnosis experts. Experiments on two benchmarks and various baselines demonstrate its effectiveness and transferability.
Problem
Existing radiology report-generation methods often overlook medical knowledge for pivotal areas, which requires extracting and aggregating knowledge at multiple granularities.
Source: §I · p. 1Method
EMGKT combines global medical CLIP embeddings, fine-grained teacher-to-student knowledge distillation, and disease-diagnosis experts initialized with disease text embeddings.
Source: §I · pp. 1–2Results
Experiments on two widely used benchmarks and various baselines demonstrate EMGKT's effectiveness and transferability.
Source: §I · p. 2, §VI · p. 8Takeaways & Limitations
EMGKT is described as compact and orthogonal to existing radiology report-generation networks, and can be applied to various baselines.
Source: §I · p. 2
Abstract
from arXiv · showhide
Radiology report generation can automatically generate clinical descriptions from X-ray images, thereby significantly improving the efficiency of radiologists. This task is challenging because it requires medical knowledge to accurately identify diseases and describe them in a professional manner. However, existing methods often overlook the importance of enhancing medical knowledge in describing pivotal areas, a capability that requires models to effectively extract and aggregate knowledge at multiple levels of granularity. Accordingly, we herein propose a novel and compact Efficient Multi-Granularity Knowledge Transfer (\textbf{EMGKT}) method to address the above issues. First, we encode global knowledge embeddings using a medical vision-language model, which provides contextual medical knowledge. Moreover, we devise a novel Fine-Grained Knowledge Distillation (FGKD) training task which efficiently extract fine-grained knowledge. Specifically, the FGKD training task contains teacher embeddings and student embeddings. Teacher embeddings are encoded using extra priors; while student embeddings are learned from the teacher embeddings through knowledge distillation. During inference, the student embeddings are used to enhance fine-grained knowledge while the teacher embeddings are discarded, resulting in negligible computational costs and no need for extra priors. Finally, we further develop a mixture of disease diagnosis expert classifiers to enhance knowledge extraction. The classifiers are initialized using disease embeddings and are modeled as different experts to address various granularity features. Notably, \textbf{EMGKT} can be efficiently applied to most existing methods. Extensive experiments are conducted on two widely-used public datasets and various baselines, which demonstrates the effectiveness and transferability of \textbf{EMGKT}.
I. INTRODUCTION
Radiology report generation must identify disease and describe it professionally, yet existing knowledge-transfer methods overlook pivotal organ and lesion areas. EMGKT combines global context, fine-grained knowledge distillation, and disease-diagnosis experts to transfer multi-granularity knowledge.
I. INTRODUCTION
- Existing knowledge-transfer methods overlook pivotal organ and lesion areas, while discriminative regions at varying scales require multi-granularity knowledge extraction and aggregation.Detection-based approaches have attempted region selection and fusion, but increase model complexity and reduce knowledge-transfer efficiency.Source: §I · p. 1
- EMGKT encodes global knowledge with a medical CLIP model and learns fine-grained student embeddings by distilling teacher embeddings built with supplementary region and phrase priors.At inference, teacher embeddings are discarded, so the method incurs negligible computational costs and needs no additional priors.Source: §I · pp. 1–2
- The authors describe EMGKT as the first multi-granularity knowledge-transfer approach for radiology report generation, with FGKD for fine-grained knowledge encoding.They report that it is compact, orthogonal to existing networks, and improves report-generation performance across various baselines.Source: §I · p. 2
- EMGKT adds a Mixture of Disease Diagnosis Experts classifier built on detailed disease-symptom descriptions and the Mixture of Experts approach.The method integrates global image context with knowledge from discriminative areas.Source: §I · p. 2, Figure 1 · p. 2
A. Image Caption Generation
The related work covers image captioning, radiology report generation, and knowledge distillation. It situates EMGKT among approaches to image-text alignment, fine-grained region features, and transfer of learned representations.
A. Image Caption Generation
- Image-captioning research uses attention to focus on salient objects, while vision-language pretraining such as BLIP supports integration of visual and textual information.The passage presents image captioning as a related area of computer vision.Source: A. Image Caption Generation · p. 2
B. Radiology Report Generation
- Radiology report generation methods address image-text misalignment, extract fine-grained regional features, or incorporate medical knowledge into clinical sentence generation.Detection frameworks for important regions can increase model complexity and overlook knowledge-transfer efficiency; existing knowledge-based methods also overlook multi-granularity knowledge.Source: B. Radiology Report Generation · pp. 2–3
C. Knowledge Distillation
- Knowledge distillation methods transfer knowledge through outputs, intermediate features, or relationships, but are typically designed for generic recognition tasks rather than medical region-level detail.The paper positions its fine-grained distillation approach as a way to transfer multi-granularity knowledge for global and local disease-related representations.Source: C. Knowledge Distillation · p. 3
III. METHOD
The baseline encodes X-ray images into image features and uses a text decoder to predict the next word from those features and previously generated words. EMGKT adds a multi-granularity knowledge encoder and disease-diagnosis experts.
III. METHOD
- The baseline encodes each input image as image features, then feeds those features and previous-word embeddings to a text decoder to produce next-word probabilities.The next word is predicted by applying softmax to the decoder's output.Source: §III · pp. 3–4
- EMGKT consists of a Multi-granularity Knowledge Encoder (MgKE) and a Mixture of Disease Diagnosis Experts (MoD2E).The method is presented as applicable to existing radiology report generation methods.Source: §III · pp. 3–4
A. Multi-Granularity Knowledge Encoder
The Multi-granularity Knowledge Encoder combines global medical context with fine-grained region knowledge. It distills teacher embeddings constructed from region priors into student embeddings used for report generation at inference.
A. Multi-Granularity Knowledge Encoder
- The encoder combines global knowledge embeddings Kg with fine-grained region embeddings Kf to represent different granularities.Global embeddings are extracted from X-ray images using a medical CLIP image encoder pretrained on image-report pairs.Source: A. Multi-Granularity Knowledge Encoder · p. 4
- Teacher embeddings are constructed using region bounding-box priors from Chest ImaGenome, which can also be obtained using large foundation models.The teacher embeddings use vision and text encoders with transformer attention modules.Source: A. Multi-Granularity Knowledge Encoder · p. 4
- The teacher feature stream augments image features with teacher embeddings, while a distillation task extracts fine-grained knowledge from global knowledge.The distillation loss uses L2 loss and a stop-gradient operator.Source: A. Multi-Granularity Knowledge Encoder · p. 4
- Only the student feature stream is used for report generation at inference; teacher embeddings are constructed for training and discarded afterward.The paper states that discarding teacher embeddings avoids extra priors and incurs negligible inference computation.Source: A. Multi-Granularity Knowledge Encoder · p. 4
B. Mixture of Disease Diagnosis Experts
MoD2E addresses disease-classifier initialization and granularity by using disease descriptions to initialize classifiers and modeling classifiers as experts for different feature types. Their classification scores are fused using a mixture-of-experts approach.
B. Mixture of Disease Diagnosis Experts
- Existing disease classifiers are typically randomly initialized and optimized with image-based labels, whereas disease identification requires medical expert knowledge.The section frames MoD2E as an approach to enhance multi-granularity knowledge extraction.Source: B. Mixture of Disease Diagnosis Experts · pp. 4–5
- MoD2E uses detailed disease descriptions generated by a large language model to initialize disease classifiers as text embeddings.The descriptions are generated from medical prompts for each disease.Source: B. Mixture of Disease Diagnosis Experts · p. 5
- The classifiers are modeled as distinct diagnosis experts for varying granularity and modality features, with classification scores fused using a mixture-of-experts approach.The expert functions perform weight assignment and disease classification.Source: B. Mixture of Disease Diagnosis Experts · p. 5
C. Training Objective
The training objective combines language-modeling and disease-classification losses. Cross-entropy supervises disease predictions from teacher and student feature streams using ground-truth labels.
C. Training Objective
- Language-modeling loss Llm optimizes report generation for the teacher and student knowledge-embedding streams.The paper follows previous works for this loss.Source: C. Training Objective · p. 5
- Cross-entropy loss Lce supervises disease-classification predictions from teacher and student streams against ground-truth disease labels.Ct and Cs are the predicted disease-logit probabilities for the two streams.Source: C. Training Objective · p. 5
- The overall loss function is formulated with α = 1.0 following, while γ and β are empirically set to 10.0 and 8.0, respectively.Source: C. Training Objective · p. 5
A. Dataset and Evaluation Metrics
The experiments use IU X-ray and MIMIC-CXR, applying consistent baseline data processing. Evaluation combines report-generation metrics with clinical disease-classification metrics.
A. Dataset and Evaluation Metrics
- IU X-ray contains 7,470 images and 3,955 reports, while MIMIC-CXR contains 377,110 images and 227,835 reports.The authors use baseline data-processing methods for consistency and fairness.Source: A. Dataset and Evaluation Metrics · p. 5
- Report generation is evaluated with BLEU, METEOR, and ROUGE-L, alongside clinical efficacy precision, recall, and F1-score.Models trained on MIMIC-CXR are also evaluated on the entire IU X-ray dataset.Source: A. Dataset and Evaluation Metrics · p. 5
B. Experiments Results and Analyses
EMGKT improves report-generation and clinical-efficacy results across benchmarks and baselines, with especially strong clinical results in cross-dataset evaluation on IU X-Ray. Ablations examine its components and knowledge-encoder and expert-classifier designs.
B. Experiments Results and Analyses
- EMGKT improves report-generation performance in cross-dataset evaluation on IU X-Ray, without fine-tuning or region-location and region-description annotations.Models trained on MIMIC-CXR are directly evaluated on IU X-Ray.Source: B. Experiments Results and Analyses · pp. 5–6
- EMGKT improves all NLG metrics across three baselines, and its disease-classification results improve across baseline models.The comparison includes R2GenCMN, PromptMRG, and REVTAF-base.Source: B. Experiments Results and Analyses · pp. 5–6
- On MIMIC-CXR, EMGKT leads BLEU-1 through BLEU-4 and all CE metrics over REVTAF, while its average score is 0.398 versus REVTAF’s 0.397.REVTAF scores higher on METEOR (0.199 vs. 0.198) and ROUGE-L (0.336 vs. 0.334).Source: B. Experiments Results and Analyses · pp. 6–7, Table III · p. 7
- On IU X-Ray, EMGKT achieves the highest average score, 0.260 versus REVTAF’s 0.251; METEOR ties at 0.176, and CE gains are 1.7%, 2.0%, and 2.0%.The CE gains are in Precision, Recall, and F1, respectively.Source: Table IV · p. 7, B. Experiments Results and Analyses · p. 7
- Using both MgKE and MoD2E improves baseline BLEU-1 by 1.40%, while both components introduce negligible inference overhead.The reported BLEU-1 gains are 1.0% with MgKE alone and 0.50% with MoD2E alone.Source: Table V · p. 7, B. Experiments Results and Analyses · p. 8
- Adding global knowledge embeddings enhances all report-generation metrics, with further improvements from fine-grained visual and textual embeddings.In the expert-classifier ablation, disease-description-based CLIP embeddings improve report-generation metrics, and MoE further improves all of them.Source: Table VI · p. 7, B. Experiments Results and Analyses · p. 8
V. QUALITATIVE ANALYSIS
The qualitative comparison attributes baseline weaknesses to incorrect identification and presentation of important regions; EMGKT improves reports by pinpointing pivotal regions and refining their descriptions.
V. QUALITATIVE ANALYSIS
- EMGKT’s examples show more precise identification of pivotal regions and refined descriptions than the baseline.The passage attributes baseline suboptimal performance primarily to incorrect identification and presentation of important regions.Source: §V · p. 8
VI. CONCLUSION
The conclusion presents EMGKT as a two-component approach: MgKE extracts multi-granularity embeddings using CLIP and knowledge distillation, while MoD2E uses disease-text embeddings to initialize expert classifiers. Experiments support effectiveness and transferability across two benchmarks and multiple baselines.
VI. CONCLUSION
- EMGKT combines MgKE, which extracts multi-granularity embeddings using CLIP and knowledge distillation, with MoD2E, which uses disease-text embeddings to initialize classifiers and MoE for different granularity features.The conclusion reports experiments on two benchmarks and various baselines as evidence of effectiveness and transferability.Source: §VI · p. 8