Yi Lin 0009

dblp:42/5120-9 · DBLP profile ↗
← Back
30ranked-venue papers
9as first author
26since 2021 · last 2026
0000-0002-7635-2518ORCID · conflict

Domains — the database's venue-derived domains; a paper can count in several

Applied, interdisciplinary, general and emerging computing · 24 · 8 first-author · 21 since 2021Graphics, computer vision, multimedia, augmented reality and games · 13 · 4 first-author · 11 since 2021Artificial intelligence and machine learning · 6 · 1 first-author · 5 since 2021
YearPublicationVenuePosition
2026 A Disease-Aware Dual-Stage Framework for Chest X-ray Report Generation
abstract
Radiology report generation from chest X-rays is an important task in artificial intelligence with the potential to greatly reduce radiologists' workload and shorten patient wait times. Despite recent advances, existing approaches often lack sufficient disease-awareness in visual representations and adequate vision-language alignment to meet the specialized requirements of medical image analysis. As a result, these models usually overlook critical pathological features on chest X-rays and struggle to generate clinically accurate reports. To address these limitations, we propose a novel dual-stage disease-aware framework for chest X-ray report generation. In Stage 1, our model learns Disease-Aware Semantic Tokens (DASTs) corresponding to specific pathology categories through cross-attention mechanisms and multi-label classification, while simultaneously aligning vision and language representations via contrastive learning. In Stage 2, we introduce a Disease-Visual Attention Fusion (DVAF) module to integrate disease-aware representations with visual features, along with a Dual-Modal Similarity Retrieval (DMSR) mechanism that combines visual and disease-specific similarities to retrieve relevant exemplars, providing contextual guidance during report generation. Extensive experiments on benchmark datasets (i.e., CheXpert Plus, IU X-ray, and MIMIC-CXR) demonstrate that our disease-aware framework achieves state-of-the-art performance in chest X-ray report generation, with significant improvements in clinical accuracy and linguistic quality.
Puzhen Wu, Hexin Dong, Yi Lin 0009, Yihao Ding, Yifan Peng 0002
AAAI3
2026 Learning with less supervision: A survey of label-efficient learning for medical image analysis
Cheng Jin 0003, Zhengrui Guo, Yi Lin 0009, Luyang Luo, Hao Chen 0011
Medical Image Anal.3
2026 LLM-Driven Medical Report Generation via Communication-Efficient Heterogeneous Federated Learning
abstract
Large Language Models (LLMs) have demonstrated significant potential in Medical Report Generation (MRG), yet their development requires large amounts of medical image-report pairs, which are commonly scattered across multiple centers. Centralizing these data is exceptionally challenging due to privacy regulations, thereby impeding model development and broader adoption of LLM-driven MRG models. To address this challenge, we present FedMRG, the first framework that leverages Federated Learning (FL) to enable privacy-preserving, multi-center development of LLM-driven MRG models, specifically designed to overcome the critical challenge of communication-efficient LLM training under multi-modal data heterogeneity. To start with, our framework tackles the fundamental challenge of communication overhead in federated LLM tuning by employing low-rank factorization to efficiently decompose parameter updates, significantly reducing gradient transmission costs and making LLM-driven MRG feasible in bandwidth-constrained FL settings. Furthermore, we observed the dual heterogeneity in MRG under the FL scenario: varying image characteristics across medical centers, as well as diverse reporting styles and terminology preferences. To address the data heterogeneity, we further enhance FedMRG with (1) client-aware contrastive learning in the MRG encoder, coupled with diagnosis-driven prompts, which capture both globally generalizable and locally distinctive features while maintaining diagnostic accuracy; and (2) a dual-adapter mutual boosting mechanism in the MRG decoder that harmonizes generic and specialized adapters to address variations in reporting styles and terminology. Through extensive evaluation of our established FL-MRG benchmark, we demonstrate the generalizability and adaptability of FedMRG, underscoring its potential in harnessing multi-center data and generating clinically accurate reports while maintaining communication efficiency.
Haoxuan Che, Haibo Jin, Zhengrui Guo, Yi Lin 0009, Cheng Jin 0003, Hao Chen 0011
IEEE Trans. Medical Imaging4
2025 Corrigendum to "Detection and analysis of cerebral aneurysms based on X-ray rotational angiography - the CADA 2020 challenge" [Medical Image Analysis, April 2022, Volume 77, 102333]
Matthias Ivantsits, Leonid Goubergrits, Jan-Martin Kuhnigk, Markus Hüllebrand, Jan Brüning, Tabea Kossen, Boris Pfahringer, Jens Schaller, Andreas Spuler, Titus Kühne, Yizhuan Jia, Xuesong Li 0003, Suprosanna Shit, Bjoern Menze, Ziyu Su, Jun Ma 0016, Ziwei Nie, Kartik Jain, Yi Lin 0009, Anja Hennemuth
Medical Image Anal.20
2025 Rethinking boundary detection in deep learning-based medical image segmentation
Yi Lin 0009, Kwang-Ting Cheng, Hao Chen 0011
Medical Image Anal.1
2025 Boosting Convolution With Efficient MLP-Permutation for Volumetric Medical Image Segmentation
abstract
Recently, the advent of Vision Transformer (ViT) has brought substantial advancements in 3D benchmarks, particularly in 3D volumetric medical image segmentation (Vol-MedSeg). Concurrently, multi-layer perceptron (MLP) network has regained popularity among researchers due to their comparable results to ViT, albeit with the exclusion of the resource-intensive self-attention module. In this work, we propose a novel permutable hybrid network for Vol-MedSeg, named PHNet, which capitalizes on the strengths of both convolution neural networks (CNNs) and MLP. PHNet addresses the intrinsic anisotropy problem of 3D volumetric data by employing a combination of 2D and 3D CNNs to extract local features. Besides, we propose an efficient multi-layer permute perceptron (MLPP) module that captures long-range dependence while preserving positional information. This is achieved through an axis decomposition operation that permutes the input tensor along different axes, thereby enabling the separate encoding of the positional information. Furthermore, MLPP tackles the resolution sensitivity issue of MLP in Vol-MedSeg with a token segmentation operation, which divides the feature into smaller tokens and processes them individually. Extensive experimental results validate that PHNet outperformed the state-of-the-art methods with lower computational costs on the widely-used yet challenging COVID-19-20, Synapse, LiTS and MSD BraTS benchmarks. The ablation study also demonstrated the effectiveness of PHNet in harnessing the strengths of both CNNs and MLP. The code is available on Github: https://github.com/xiaofang007/PHNet.
Yi Lin 0009, Kwang-Ting Cheng, Hao Chen 0011
IEEE Trans. Medical Imaging1
2025 Merging Context Clustering With Visual State Space Models for Medical Image Segmentation
abstract
Medical image segmentation demands the aggregation of global and local feature representations, posing a challenge for current methodologies in handling both long-range and short-range feature interactions. Recently, vision mamba (ViM) models have emerged as promising solutions for addressing model complexities by excelling in long-range feature iterations with linear complexity. However, existing ViM approaches overlook the importance of preserving short-range local dependencies by directly flattening spatial tokens and are constrained by fixed scanning patterns that limit the capture of dynamic spatial context information. To address these challenges, we introduce a simple yet effective method named context clustering ViM (CCViM), which incorporates a context clustering module within the existing ViM models to segment image tokens into distinct windows for adaptable local clustering. Our method effectively combines long-range and short-range feature interactions, thereby enhancing spatial contextual representations for medical image segmentation tasks. Extensive experimental evaluations on diverse public datasets, i.e., Kumar, CPM17, ISIC17, ISIC18, and Synapse, demonstrate the superior performance of our method compared to current state-of-the-art methods. Our code can be found at https://github.com/zymissy/CCViM.
Yi Lin 0009, Jinhui Tang 0001
IEEE Trans. Medical Imaging3
2024 PromptMRG: Diagnosis-Driven Prompts for Medical Report Generation
abstract
Automatic medical report generation (MRG) is of great research value as it has the potential to relieve radiologists from the heavy burden of report writing. Despite recent advancements, accurate MRG remains challenging due to the need for precise clinical understanding and disease identification. Moreover, the imbalanced distribution of diseases makes the challenge even more pronounced, as rare diseases are underrepresented in training data, making their diagnosis unreliable. To address these challenges, we propose diagnosis-driven prompts for medical report generation (PromptMRG), a novel framework that aims to improve the diagnostic accuracy of MRG with the guidance of diagnosis-aware prompts. Specifically, PromptMRG is based on encoder-decoder architecture with an extra disease classification branch. When generating reports, the diagnostic results from the classification branch are converted into token prompts to explicitly guide the generation process. To further improve the diagnostic accuracy, we design cross-modal feature enhancement, which retrieves similar reports from the database to assist the diagnosis of a query image by leveraging the knowledge from a pre-trained CLIP. Moreover, the disease imbalanced issue is addressed by applying an adaptive logit-adjusted loss to the classification branch based on the individual learning status of each disease, which overcomes the barrier of text decoder's inability to manipulate disease distributions. Experiments on two MRG benchmarks show the effectiveness of the proposed method, where it obtains state-of-the-art clinical efficacy performance on both datasets.
Haibo Jin, Haoxuan Che, Yi Lin 0009, Hao Chen 0011
AAAI3
2024 Aligning Medical Images with General Knowledge from Large Language Models
Yi Lin 0009, Kwang-Ting Cheng, Hao Chen 0011
MICCAI (10)2
2024 Revisiting Deep Ensemble Uncertainty for Enhanced Medical Anomaly Detection
Yi Lin 0009, Kwang-Ting Cheng, Hao Chen 0011
MICCAI (6)2
2024 Iterative Online Image Synthesis via Diffusion Model for Imbalanced Classification
Shuhan Li, Yi Lin 0009, Hao Chen 0011, Kwang-Ting Cheng
MICCAI (5)2
2024 TAKT: Target-Aware Knowledge Transfer for Whole Slide Image Classification
Conghao Xiong, Yi Lin 0009, Hao Chen 0011, Hao Zheng 0008, Dong Wei 0004, Yefeng Zheng 0001, Joseph J. Y. Sung, Irwin King
MICCAI (4)2
2024 Triplet-branch network with contrastive prior-knowledge embedding for disease grading
Yuexiang Li, Yawen Huang, Jingxin Liu 0005, Yi Lin 0009, Dong Wei 0004, Qirui Zhang 0004, Kai Ma 0002, Guangming Lu 0001, Yefeng Zheng 0001
Artif. Intell. Medicine6
2024 CAE-GReaT: Convolutional-Auxiliary Efficient Graph Reasoning Transformer for Dense Image Predictions
Yi Lin 0009, Jinhui Tang 0001, Kwang-Ting Cheng
Int. J. Comput. Vis.2
2024 LENAS: Learning-Based Neural Architecture Search and Ensemble for 3-D Radiotherapy Dose Prediction
abstract
Radiation therapy treatment planning requires balancing the delivery of the target dose while sparing normal tissues, making it a complex process. To streamline the planning process and enhance its quality, there is a growing demand for knowledge-based planning (KBP). Ensemble learning has shown impressive power in various deep learning tasks, and it has great potential to improve the performance of KBP. However, the effectiveness of ensemble learning heavily depends on the diversity and individual accuracy of the base learners. Moreover, the complexity of model ensembles is a major concern, as it requires maintaining multiple models during inference, leading to increased computational cost and storage overhead. In this study, we propose a novel learning-based ensemble approach named LENAS, which integrates neural architecture search with knowledge distillation for 3-D radiotherapy dose prediction. Our approach starts by exhaustively searching each block from an enormous architecture space to identify multiple architectures that exhibit promising performance and significant diversity. To mitigate the complexity introduced by the model ensemble, we adopt the teacher-student paradigm, leveraging the diverse outputs from multiple learned networks as supervisory signals to guide the training of the student network. Furthermore, to preserve high-level semantic information, we design a hybrid loss to optimize the student network, enabling it to recover the knowledge embedded within the teacher networks. The proposed method has been evaluated on two public datasets: 1) OpenKBP and 2) AIMIS. Extensive experimental results demonstrate the effectiveness of our method and its superior performance to the state-of-the-art methods. Code: github.com/hust-linyi/LENAS.
Yi Lin 0009, Hao Chen 0011, Xin Yang 0008, Kai Ma 0002, Yefeng Zheng 0001, Kwang-Ting Cheng
IEEE Trans. Cybern.1
2024 Adaptive Fusion of Deep Learning With Statistical Anatomical Knowledge for Robust Patella Segmentation From CT Images
abstract
Kneeosteoarthritis (KOA), as a leading joint disease, can be decided by examining the shapes of patella to spot potential abnormal variations. To assist doctors in the diagnosis of KOA, a robust automatic patella segmentation method is highly demanded in clinical practice. Deep learning methods, especially convolutional neural networks (CNNs) have been widely applied to medical image segmentation in recent years. Nevertheless, poor image quality and limited data still impose challenges to segmentation via CNNs. On the other hand, statistical shape models (SSMs) can generate shape priors which give anatomically reliable segmentation to varying instances. Thus, in this work, we propose an adaptive fusion framework, explicitly combining deep neural networks and anatomical knowledge from SSM for robust patella segmentation. Our adaptive fusion framework will accordingly adjust the weight of segmentation candidates in fusion based on their segmentation performance. We also propose a voxel-wise refinement strategy to make the segmentation of CNNs more anatomically correct. Extensive experiments and thorough assessment have been conducted on various mainstream CNN backbones for patella segmentation in low-data regimes, which demonstrate that our framework can be flexibly attached to a CNN model, significantly improving its performance when labeled training data are limited and input image data are of poor quality.
Tianshu Jiang, Yi Lin 0009, Lok-Chun Chan, Ping-Keung Chan, Chun-Yi Wen, Hao Chen 0011
IEEE J. Biomed. Health Informatics3
2024 BoNuS: Boundary Mining for Nuclei Segmentation With Partial Point Labels
abstract
Nuclei segmentation is a fundamental prerequisite in the digital pathology workflow. The development of automated methods for nuclei segmentation enables quantitative analysis of the wide existence and large variances in nuclei morphometry in histopathology images. However, manual annotation of tens of thousands of nuclei is tedious and time-consuming, which requires significant amount of human effort and domain-specific expertise. To alleviate this problem, in this paper, we propose a weakly-supervised nuclei segmentation method that only requires partial point labels of nuclei. Specifically, we propose a novel boundary mining framework for nuclei segmentation, named BoNuS, which simultaneously learns nuclei interior and boundary information from the point labels. To achieve this goal, we propose a novel boundary mining loss, which guides the model to learn the boundary information by exploring the pairwise pixel affinity in a multiple-instance learning manner. Then, we consider a more challenging problem, i.e., partial point label, where we propose a nuclei detection module with curriculum learning to detect the missing nuclei with prior morphological knowledge. The proposed method is validated on three public datasets, MoNuSeg, CPM, and CoNIC datasets. Experimental results demonstrate the superior performance of our method to the state-of-the-art weakly-supervised nuclei segmentation methods. Code: https://github.com/hust-linyi/bonus.
Yi Lin 0009, Kwang-Ting Cheng, Hao Chen 0011
IEEE Trans. Medical Imaging1
2024 Adversarial Medical Image With Hierarchical Feature Hiding
abstract
Deep learning based methods for medical images can be easily compromised by adversarial examples (AEs), posing a great security flaw in clinical decision-making. It has been discovered that conventional adversarial attacks like PGD which optimize the classification logits, are easy to distinguish in the feature space, resulting in accurate reactive defenses. To better understand this phenomenon and reassess the reliability of the reactive defenses for medical AEs, we thoroughly investigate the characteristic of conventional medical AEs. Specifically, we first theoretically prove that conventional adversarial attacks change the outputs by continuously optimizing vulnerable features in a fixed direction, thereby leading to outlier representations in the feature space. Then, a stress test is conducted to reveal the vulnerability of medical images, by comparing with natural images. Interestingly, this vulnerability is a double-edged sword, which can be exploited to hide AEs. We then propose a simple-yet-effective hierarchical feature constraint (HFC), a novel add-on to conventional white-box attacks, which assists to hide the adversarial feature in the target feature distribution. The proposed method is evaluated on three medical datasets, both 2D and 3D, with different modalities. The experimental results demonstrate the superiority of HFC,i.e., it bypasses an array of state-of-the-art adversarial medical AE detectors more efficiently than competing adaptive attacks1, which reveals the deficiencies of medical reactive defense and allows to develop more robust defenses in future.
Qingsong Yao, Zecheng He, Yuexiang Li, Yi Lin 0009, Kai Ma 0002, Yefeng Zheng 0001, Shaohua Kevin Zhou
IEEE Trans. Medical Imaging4
2023 Few Shot Medical Image Segmentation with Cross Attention Transformer
Yi Lin 0009, Kwang-Ting Cheng, Hao Chen 0011
MICCAI (2)1
2023 Deep learning for computational cytology: A survey
Hao Jiang 0028, Yanning Zhou 0001, Yi Lin 0009, Ronald C. K. Chan, Jiang Liu 0001, Hao Chen 0011
Medical Image Anal.3
2023 Nuclei segmentation with point annotations from pathology images via self-supervised learning and co-training
Yi Lin 0009, Zhiyong Qu, Hao Chen 0011, Zhongke Gao, Yuexiang Li, Kai Ma 0002, Yefeng Zheng 0001, Kwang-Ting Cheng
Medical Image Anal.1
2022 InsMix: Towards Realistic Generative Data Augmentation for Nuclei Instance Segmentation
Yi Lin 0009, Kwang-Ting Cheng, Hao Chen 0011
MICCAI (2)1
2022 Detection and analysis of cerebral aneurysms based on X-ray rotational angiography - the CADA 2020 challenge
Matthias Ivantsits, Leonid Goubergrits, Jan-Martin Kuhnigk, Markus Hüllebrand, Jan Brüning, Tabea Kossen, Boris Pfahringer, Jens Schaller, Andreas Spuler, Titus Kühne, Yizhuan Jia, Xuesong Li 0003, Suprosanna Shit, Bjoern Menze, Ziyu Su, Jun Ma 0016, Ziwei Nie, Kartik Jain, Yi Lin 0009, Anja Hennemuth
Medical Image Anal.20
2021 Triplet-Branch Network with Prior-Knowledge Embedding for Fatigue Fracture Grading
Yuexiang Li, Yi Lin 0009, Dong Wei 0004, Qirui Zhang 0004, Kai Ma 0002, Guangming Lu 0001, Yefeng Zheng 0001
MICCAI (5)4
2021 Seg4Reg+: Consistency Learning Between Spine Segmentation and Cobb Angle Regression
Yi Lin 0009, Luyan Liu, Kai Ma 0002, Yefeng Zheng 0001
MICCAI (5)1
2021 A Hierarchical Feature Constraint to Camouflage Medical Adversarial Attacks
Qingsong Yao, Zecheng He, Yi Lin 0009, Kai Ma 0002, Yefeng Zheng 0001, Shaohua Kevin Zhou
MICCAI (3)3
2020 Semi-supervised mp-MRI data synthesis with StitchLayer and auxiliary distance maximization
Zhiwei Wang 0002, Yi Lin 0009, Kwang-Ting Cheng, Xin Yang 0008
Medical Image Anal.2
2020 Bi-Modality Medical Image Synthesis Using Semi-Supervised Sequential Generative Adversarial Networks
abstract
In this paper, we propose a bi-modality medical image synthesis approach based on sequential generative adversarial network (GAN) and semi-supervised learning. Our approach consists of two generative modules that synthesize images of the two modalities in a sequential order. A method for measuring the synthesis complexity is proposed to automatically determine the synthesis order in our sequential GAN. Images of the modality with a lower complexity are synthesized first, and the counterparts with a higher complexity are generated later. Our sequential GAN is trained end-to-end in a semi-supervised manner. In supervised training, the joint distribution of bi-modality images are learned from real paired images of the two modalities by explicitly minimizing the reconstruction losses between the real and synthetic images. To avoid overfitting limited training images, in unsupervised training, the marginal distribution of each modality is learned based on unpaired images by minimizing the Wasserstein distance between the distributions of real and fake images. We comprehensively evaluate the proposed model using two synthesis tasks based on three types of evaluate metrics and user studies. Visual and quantitative results demonstrate the superiority of our method to the state-of-the-art methods, and reasonable visual quality and clinical significance. Code is made publicly available at https://github.com/hust- linyi/Multimodal-Medical-Image-Synthesis.
Xin Yang 0008, Yi Lin 0009, Zhiwei Wang 0002, Xin Li 0001, Kwang-Ting Cheng
IEEE J. Biomed. Health Informatics2
2019 Automated Pulmonary Embolism Detection from CTPA Images Using an End-to-End Convolutional Neural Network
Yi Lin 0009, Jianchao Su, Jingen Liu, Kwang-Ting Cheng, Xin Yang 0008
MICCAI (4)1
2018 StitchAD-GAN for Synthesizing Apparent Diffusion Coefficient Images of Clinically Significant Prostate Cancer
Zhiwei Wang 0002, Yi Lin 0009, Chunyuan Liao, Kwang-Ting Cheng, Xin Yang 0008
BMVC2