Hengfei Cui

dblp:205/8151 · DBLP profile ↗
← Back
22ranked-venue papers
6as first author
21since 2021 · last 2026
0000-0001-8625-2521ORCID · verified

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

Applied, interdisciplinary, general and emerging computing · 11 · 3 first-author · 11 since 2021Graphics, computer vision, multimedia, augmented reality and games · 9 · 9 since 2021Artificial intelligence and machine learning · 8 · 3 first-author · 7 since 2021
YearPublicationVenuePosition
2026 MRFMA: A hybrid paradigm integrating multi-receptive field network with mediator attention for 3D multi-organ segmentation
Hengfei Cui, Jiatong Li 0006, Dianrong Du, Yanning Zhang 0001, Yong Xia 0001
Expert Syst. Appl.1
2026 Unsupervised domain adaptation for cardiac MRI segmentation via adversarial learning in latent space
Hengfei Cui, Yong Xia 0001
Pattern Recognit.2
2025 Transfer Attention-Guided Multi-Receptive Field Network for Multi-Modality Cardiac Image Segmentation
abstract
Existing whole heart segmentation algorithms usually combine 3D Convolutional Neural Networks (3D CNNs) with Transformers, for the purpose of capturing local and global features. However, traditional CNNs with a fixed size of receptive field cannot capture long-range contextual information. Transformers have been widely used to establish dependencies on global information, despite this, they greatly increase the computational complexity. To mitigate these challenges, we propose a hybrid paradigm, called Transfer Attention-Guided MultiReceptive Field Network (TAMRNet), to boost the representation quality for multi-modality cardiac image segmentation. In TAMRNet, the novel adaptive-scale depthwise convolution module adeptly preserves the inherent inductive biases of convolution while concurrently amplifying the network's ability to establish dependencies on long-range contextual information. Besides, a novel attention mechanism called Transfer Attention is developed to establish dependencies on global information. Transfer Attention avoids the direct similarity calculation of$Q$and$K$by introducing the Transfer tokens, and thus dramatically decreases the computational cost. The proposed TAMRNet is tested on the MM-WHS 2017 challenge dataset, achieving the average Dice scores of 93.7 % and 82.2 % on the CT and MRI datasets respectively. Extensive experimental results prove that our proposed method achieves superior performances in comparison with state-of-the-art methods.
Jiatong Li 0006, Hengfei Cui, Dianrong Du, Geng Chen 0001, Yong Xia 0001
BIBM2
2025 SMF-Net: Unlocking Multimodal Insights for Enhanced Stroke Lesion Segmentation
Meklit Mesfin Atlaw, Geng Chen 0001, Xuyun Wen, Hengfei Cui, Yong Xia 0001
MICCAI (3)5
2025 Personalized Federated Side-Tuning for Medical Image Classification
Jiayi Chen 0006, Benteng Ma, Yongsheng Pan, Bin Pu, Hengfei Cui, Yong Xia 0001
MICCAI (14)5
2025 CAUDA-MI: Cross Attention-Guided Unsupervised Domain Adaptation with Mutual Information for Cardiac MRI Segmentation
Dianrong Du, Hengfei Cui, Jiatong Li 0006, Yong Xia 0001
MICCAI (6)2
2025 Towards Accurate Left Atrium and Scar Segmentation from LGE MRI with Boundary Loss Constrained Multi-Attention U-Net
Hengfei Cui, Jiatong Li 0006, Dianrong Du, Geng Chen 0001, Yong Xia 0001
PRCV (14)2
2025 Mixture-attention Siamese transformer for video polyp segmentation
Geng Chen 0001, Junqing Yang, Xiaozhou Pu, Ge-Peng Ji, Huan Xiong, Yongsheng Pan, Hengfei Cui, Yong Xia 0001
Artif. Intell. Medicine7
2025 Active Learning Based on Temporal Difference of Gradient Flow in Thoracic Disease Diagnosis
abstract
Given the significant advancements in thoracic disease diagnosis due to deep learning, there is a reliance on the availability of numerous annotated samples, which, however, can hardly be guaranteed due to the resource-intensive nature of medical image annotation. Active learning has been introduced to mitigate annotation costs by selecting a subset of uncertain samples for annotation and training. Existing active learning methods encounter two primary challenges: 1) overlooking the impact of samples on the dynamics of model training during data selection, and 2) suffering from high costs of data evaluation and selection. To tackle both issues, we propose a novel metric called Temporal Difference of Gradient Flow (TDGF) for data selection in active learning. Each round of active learning involves three steps: model training, data selection, and data annotation. First, we train a target model, a proxy model, and a historical proxy model on the labeled set. Second, the TDGF scores of unlabeled samples are evaluated based on the surrogate gradient flow, i.e., the TDGF w.r.t the final fully-connected layer between the proxy and historical proxy models, and top-K samples with the highest TDGF scores are selected. Third, the selected samples are annotated, and the labeled pool and unlabeled pool are updated. Comparative experiments have been conducted on two public chest radiograph datasets, i.e., ChestX-ray14 and CheXpert. Our results suggest that the proposed TDGF metric is prone to selecting hard and uncertain samples, and the use of proxy models and surrogate gradient flow substantially reduces the complexity of TDGF calculation. More importantly, the results also indicate that our TDGF-based method outperforms classical and state-of-the-art active learning methods in thoracic disease diagnosis.
Jiayi Chen 0006, Benteng Ma, Hengfei Cui, Jingfeng Zhang, Yong Xia 0001
IEEE J. Biomed. Health Informatics3
2025 P2TC: A Lightweight Pyramid Pooling Transformer-CNN Network for Accurate 3D Whole Heart Segmentation
abstract
Cardiovascular disease is a leading global cause of death, requiring accurate heart segmentation for diagnosis and surgical planning. Deep learning methods have been demonstrated to achieve superior performances in cardiac structures segmentation. However, there are still limitations in 3D whole heart segmentation, such as inadequate spatial context modeling, difficulty in capturing long-distance dependencies, high computational complexity, and limited representation of local high-level semantic information. To tackle the above problems, we propose a lightweight Pyramid Pooling Transformer-CNN (P2TC) network for accurate 3D whole heart segmentation. The proposed architecture comprises a dual encoder-decoder structure with a 3D pyramid pooling Transformer for multi-scale information fusion and a lightweight large-kernel Convolutional Neural Network (CNN) for local feature extraction. The decoder has two branches for precise segmentation and contextual residual handling. The first branch is used to generate segmentation masks for pixel-level classification based on the features extracted by the encoder to achieve accurate segmentation of cardiac structures. The second branch highlights contextual residuals across slices, enabling the network to better handle variations and boundaries. Extensive experimental results on the Multi-Modality Whole Heart Segmentation (MM-WHS) 2017 challenge dataset demonstrate that P2TC outperforms the most advanced methods, achieving the Dice scores of 92.6% and 88.1% in Computed Tomography (CT) and Magnetic Resonance Imaging (MRI) modalities respectively, which surpasses the baseline model by 1.5% and 1.7%, and achieves state-of-the-art segmentation results.
Hengfei Cui, Yifan Wang 0033, Yan Li 0129, Yanning Zhang 0001, Yong Xia 0001
IEEE J. Biomed. Health Informatics1
2024 Think Twice Before Selection: Federated Evidential Active Learning for Medical Image Analysis with Domain Shifts
abstract
Federated learning facilitates the collaborative learning of a global model across multiple distributed medical in-stitutions without centralizing data. Nevertheless, the ex-pensive cost of annotation on local clients remains an ob-stacle to effectively utilizing local data. To mitigate this issue, federated active learning methods suggest leveraging local and global model predictions to select a relatively small amount of informative local data for annotation. However, existing methods mainly focus on all local data sampled from the same domain, making them un-reliable in realistic medical scenarios with domain shifts among different clients. In this paper, we make the first at-tempt to assess the informativeness of local data derived from diverse domains and propose a novel methodology termed Federated Evidential Active Learning (FEAL) to calibrate the data evaluation under domain shift. Specif-ically, we introduce a Dirichlet prior distribution in both local and global models to treat the prediction as a distribution over the probability simplex and capture both aleatoric and epistemic uncertainties by using the Dirichlet-based evidential model. Then we employ the epistemic uncer-tainty to calibrate the aleatoric uncertainty. Afterward, we design a diversity relaxation strategy to reduce data re-dundancy and maintain data diversity. Extensive experi-ments and analysis on five real multi-center medical image datasets demonstrate the superiority of FEAL over the state-of-the-art active learning methods in federated sce-narios with domain shifts. The code will be available at https://github.com/JiayiChen815/FEAL.
Jiayi Chen 0006, Benteng Ma, Hengfei Cui, Yong Xia 0001
CVPR3
2024 FedEvi: Improving Federated Medical Image Segmentation via Evidential Weight Aggregation
Jiayi Chen 0006, Benteng Ma, Hengfei Cui, Yong Xia 0001
MICCAI (10)3
2024 DisControlFace: Adding Disentangled Control to Diffusion Autoencoder for One-shot Explicit Facial Image Editing
Haozhe Jia, Yan Li 0129, Hengfei Cui, Di Xu 0012, Yuwang Wang, Tao Yu 0007
ACM Multimedia3
2024 Toward Accurate Cardiac MRI Segmentation With Variational Autoencoder-Based Unsupervised Domain Adaptation
abstract
Accurate myocardial segmentation is crucial in the diagnosis and treatment of myocardial infarction (MI), especially in Late Gadolinium Enhancement (LGE) cardiac magnetic resonance (CMR) images, where the infarcted myocardium exhibits a greater brightness. However, segmentation annotations for LGE images are usually not available. Although knowledge gained from CMR images of other modalities with ample annotations, such as balanced-Steady State Free Precession (bSSFP), can be transferred to the LGE images, the difference in image distribution between the two modalities (i.e., domain shift) usually results in a significant degradation in model performance. To alleviate this, an end-to-end Variational autoencoder based feature Alignment Module Combining Explicit and Implicit features (VAMCEI) is proposed. We first re-derive the Kullback-Leibler (KL) divergence between the posterior distributions of the two domains as a measure of the global distribution distance. Second, we calculate the prototype contrastive loss between the two domains, bringing closer the prototypes of the same category across domains and pushing away the prototypes of different categories within or across domains. Finally, a domain discriminator is added to the output space, which indirectly aligns the feature distribution and forces the extracted features to be more favorable for segmentation. In addition, by combining CycleGAN and VAMCEI, we propose a more refined multi-stage unsupervised domain adaptation (UDA) framework for myocardial structure segmentation. We conduct extensive experiments on the MSCMRSeg 2019, MyoPS 2020 and MM-WHS 2017 datasets. The experimental results demonstrate that our framework achieves superior performances than state-of-the-art methods.
Hengfei Cui, Yan Li 0129, Yifan Wang 0033, Di Xu 0012, Lianming Wu, Yong Xia 0001
IEEE Trans. Medical Imaging1
2024 Robust Stochastic Neural Ensemble Learning With Noisy Labels for Thoracic Disease Classification
abstract
Chest radiography is the most common radiology examination for thoracic disease diagnosis, such as pneumonia. A tremendous number of chest X-rays prompt data-driven deep learning models in constructing computer-aided diagnosis systems for thoracic diseases. However, in realistic radiology practice, a deep learning-based model often suffers from performance degradation when trained on data with noisy labels possibly caused by different types of annotation biases. To this end, we present a novel stochastic neural ensemble learning (SNEL) framework for robust thoracic disease diagnosis using chest X-rays. The core idea of our method is to learn from noisy labels by constructing model ensembles and designing noise-robust loss functions. Specifically, we propose a fast neural ensemble method that collects parameters simultaneously across model instances and along optimization trajectories. Moreover, we propose a loss function that both optimizes a robust measure and characterizes a diversity measure of ensembles. We evaluated our proposed SNEL method on three publicly available hospital-scale chest X-ray datasets. The experimental results indicate that our method outperforms competing methods and demonstrate the effectiveness and robustness of our method in learning from noisy labels. Our code is available at https://github.com/hywang01/SNEL.
Hongyu Wang 0011, Hengfei Cui, Yong Xia 0001
IEEE Trans. Medical Imaging3
2023 Treasure in Distribution: A Domain Randomization Based Multi-source Domain Generalization for 2D Medical Image Segmentation
Ziyang Chen 0003, Yongsheng Pan, Yiwen Ye, Hengfei Cui, Yong Xia 0001
MICCAI (4)4
2023 DAN-NucNet: A dual attention based framework for nuclei segmentation in cancer histology images under wild clinical conditions
Ibtihaj Ahmad, Yong Xia 0001, Hengfei Cui, Zain Ul Islam
Expert Syst. Appl.3
2023 FP-DARTS: Fast parallel differentiable neural architecture search for image classification
Wenna Wang, Xiuwei Zhang 0001, Hengfei Cui, Hanlin Yin, Yanning Zhang 0001
Pattern Recognit.3
2023 An Improved Combination of Faster R-CNN and U-Net Network for Accurate Multi-Modality Whole Heart Segmentation
abstract
Detailed information of substructures of the whole heart is usually vital in the diagnosis of cardiovascular diseases and in 3D modeling of the heart. Deep convolutional neural networks have been demonstrated to achieve state-of-the-art performance in 3D cardiac structures segmentation. However, when dealing with high-resolution 3D data, current methods employing tiling strategies usually degrade segmentation performances due to GPU memory constraints. This work develops a two-stage multi-modality whole heart segmentation strategy, which adopts an improved Combination of Faster R-CNN and 3D U-Net (CFUN+). More specifically, the bounding box of the heart is first detected by Faster R-CNN, and then the original Computed Tomography (CT) and Magnetic Resonance Imaging (MRI) images of the heart aligned with the bounding box are input into 3D U-Net for segmentation. The proposed CFUN+ method redefines the bounding box loss function by replacing the previous Intersection over Union (IoU) loss with Complete Intersection over Union (CIoU) loss. Meanwhile, the integration of the edge loss makes the segmentation results more accurate, and also improves the convergence speed. The proposed method achieves an average Dice score of 91.1% on the Multi-Modality Whole Heart Segmentation (MM-WHS) 2017 challenge CT dataset, which is 5.2% higher than the baseline CFUN model, and achieves state-of-the-art segmentation results. In addition, the segmentation speed of a single heart has been dramatically improved from a few minutes to less than 6 seconds.
Hengfei Cui, Yifan Wang 0033, Yan Li 0129, Di Xu 0010, Lei Jiang 0015, Yong Xia 0001, Yanning Zhang 0001
IEEE J. Biomed. Health Informatics1
2022 Deep U-Net architecture with curriculum learning for myocardial pathology segmentation in multi-sequence cardiac magnetic resonance images
Hengfei Cui, Lei Jiang 0015, Chang Yuwen, Yong Xia 0001, Yanning Zhang 0001
Knowl. Based Syst.1
2021 Dual Attention Guided R2 U-Net Architecture for Right Ventricle Segmentation in MRI Images
Lei Jiang 0015, Hengfei Cui, Chang Yuwen, Yanning Zhang 0001
ICIG (2)2
2018 Validation of right coronary artery lumen area from cardiac computed tomography against intravascular ultrasound
Hengfei Cui, Yong Xia 0001, Yanning Zhang 0001, Liang Zhong 0001
Mach. Vis. Appl.1