Yitian Zhao

dblp:17/9876 · DBLP profile ↗
← Back
131ranked-venue papers
14as first author
88since 2021 · last 2026
0000-0003-4357-4592ORCID · conflict

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

Applied, interdisciplinary, general and emerging computing · 79 · 8 first-author · 57 since 2021Graphics, computer vision, multimedia, augmented reality and games · 64 · 6 first-author · 41 since 2021Artificial intelligence and machine learning · 33 · 3 first-author · 20 since 2021
YearPublicationVenuePosition
2026 Fundus image quality assessment in retinopathy of prematurity via multi-label graph evidential network
Donghan Wu, Wenyue Shen, Heng Li 0010, Huaying Hao, Juan Ye, Yitian Zhao
Medical Image Anal.7
2026 YoloSeg: You only label once for medical image segmentation
abstract
Acquiring pixel-level annotations for medical images is an extremely time-consuming and labor-intensive task, typically occupying the majority of the development cycle for medical image segmentation models. While existing semi-supervised methods have achieved promising results, they generally still require annotations for 10%-30% of the samples to effectively guide learning from unlabeled data, which remains a substantial burden for real-world applications. In this study, we propose YoloSeg, a novel framework for medical image segmentation under extreme label scarcity, where only a single labeled image is available. YoloSeg integrates Segment Anything Model 2 to propagate labels from the labeled image to unlabeled images, thereby expanding the labeled data pool. To address the inherent noise in pseudo-labels, we employ multi-view label propagation, decomposing pseudo-labels into consensus and divergence regions. We introduce a dual-component loss to handle these regions separately, facilitating more robust pseudo-label learning for segmentation models. Additionally, we propose a cross-patch data augmentation strategy to generate new samples with stronger semantic consistency, further enhancing the stability of training and improving model generalization. We validate our method on ten diverse medical image segmentation datasets, encompassing a wide range of segmentation targets including organs, vessels, and lesions. Experimental results show that YoloSeg achieves performance comparable to fully-supervised baselines, with an average Dice score difference of only 3.08% across all tasks, and significantly outperforms other state-of-the-art semi-supervised and one-shot methods. YoloSeg significantly improves the feasibility and cost-effectiveness of deep learning in scenarios with severely limited annotation budgets. This approach holds promise for enabling the rapid development and deployment of custom segmentation models across diverse medical centers, thereby supporting the broader adoption of intelligent medical technologies. Code is available at https://github.com/iMED-Lab/YoloSeg.
Mingen Zhang, Meng Wang 0038, Lei Mou, Jingfeng Zhang, Yitian Zhao
Medical Image Anal.6
2026 Multi-Granularity Topological Reasoning for Anatomically Consistent Vasculature Parsing
abstract
Quantitative analysis of retinal vascular morphology is vital for clinical decision-making and the investigation of systemic diseases. Central to this process is the accurate segmentation of retinal arteries and veins (A/V) from the background, a task challenged by substantial variations in vessel calibers and the presence of low-contrast or ambiguous structures in fundus images, especially in ultra-wide field imaging where peripheral distortions and large-scale anatomical variability are pronounced. These factors often lead to fragmented semantic representations and topological inconsistencies in automated segmentation outputs. To address these limitations, we propose Ultra, a multi-granularity topological reasoning network designed for precise A/V segmentation. Ultra adopts a cascaded two-stage architecture: PriorNet generates coarse, multi-scale vascular priors that provide structural guidance, while RefineNet performs topology-aware segmentation refinement. To further enforce topological coherence, we propose the neighboring pixel connectivity regularization (NICER) layer, which selectively integrates local connectivity information predicted by the proposed connectivity prediction union (CPU) module. This connectivity is employed as auxiliary supervision through a pixel-wise local connectivity loss, reinforcing structural reasoning and promoting anatomically consistent vascular topology inference. Extensive experiments on ultra-wide field fundus imaging (UWF) datasets demonstrate that Ultra achieves state-of-the-art performance in A/V segmentation and topological preservation. Moreover, Ultra generalizes well to conventional color fundus photography (CFP) datasets, underscoring its robustness and broad applicability. Code is publicly available at: https://github.com/iMED-Lab/Ultra.
Lei Mou, Yonghuai Liu, Zhuoting Xu, Hao Zhang 0113, Yalin Zheng, Jiang Liu 0001, Huazhu Fu, Yitian Zhao
IEEE Trans. Image Process.8
2026 StealthMark: Harmless and Stealthy Ownership Verification for Medical Segmentation via Uncertainty-Guided Backdoors
abstract
Annotating medical data for training AI models is often costly and limited due to the shortage of specialists with relevant clinical expertise. This challenge is further compounded by privacy and ethical concerns associated with sensitive patient information. As a result, well-trained medical segmentation models on private datasets constitute valuable intellectual property requiring robust protection mechanisms. Existing model protection techniques primarily focus on classification and generative tasks, while segmentation models-crucial to medical image analysis-remain largely underexplored. In this paper, we propose a novel, stealthy, and harmless method, StealthMark, for verifying the ownership of medical segmentation models under closed-box conditions. Our approach subtly modulates model uncertainty without altering the final segmentation outputs, thereby preserving the model's performance. To enable ownership verification, we incorporate model-agnostic explanation methods, e.g. LIME, to extract feature attributions from the model outputs. Under specific triggering conditions, these explanations reveal a distinct and verifiable watermark. We further design the watermark as a QR code to facilitate robust and recognizable ownership claims. We conducted extensive experiments across four medical imaging datasets (CMR dataset from UK Biobank, the SEG fundus dataset, the EchoNet echocardiography dataset, and the PraNet colonoscopy dataset) and five mainstream segmentation models. The results demonstrate the effectiveness, stealthiness, and harmlessness of our method on the original model's segmentation performance. For example, when applied to the SAM model, StealthMark consistently achieved attack success rates (ASR) above 95% across various datasets while maintaining less than a 1% drop in Dice and AUC scores-significantly outperforming backdoor-based watermarking methods and highlighting its strong potential for practical deployment. Our implementation code is made available at https://github.com/Qinkaiyu/StealthMark.
Qinkai Yu, Chong Zhang 0006, Gaojie Jin, Tianjin Huang, Wei Zhou 0021, Xiao-Bo Jin, Bo Huang 0012, Yitian Zhao, Gregory Yoke Hong Lip, Yalin Zheng, Aline Villavicencio, Yanda Meng
IEEE Trans. Image Process.9
2026 Super-Resolution Reconstruction of OCTA via Multi-Field-of-View Representation Learning
abstract
High-resolution Optical Coherence Tomography Angiography (OCTA) images are essential for morphological analysis and biomarker measurement of the retinal vasculature. They can also provide underlying biomarkers for the accurate analysis of eye-related diseases. The trade-off between the high resolution (HR) and large scanning field-of-view (FOV) is a long-standing problem for OCTA image instrument. A large FOV image provides more retinal information with shorter acquisition time but often suffers from low resolution (LR), high scatter noise, and poor vascular contrast. In order to obtain HR OCTA images with larger FOV, we propose a novel self-similar dynamic domain adaptation network based on cross-field-of-view representation learning. The network enables LR images (i.e., $6\times \text{6}\,\text{mm}^{2}$) to learn HR image (i.e., $3\times \text{3}\,\text{mm}^{2}$) feature representations specialized for OCTA by constructing feature mapping relations for cross-field-of-view OCTA scans. To be specific, a multiple random degradation model is proposed on HR images to generate various synthetic LR images. Further, we propose a dynamic domain adaptation framework that prompts feature dynamic alignment of the LR image reconstruction results with those of synthetic LR images. Finally, a novel self-similar supervision loss is proposed to optimize the reconstruction results from LR to HR by exploiting the similarity between vessels in different regions. Experimental results on three OCTA datasets show that the proposed method surpasses existing state-of-the-art ones, significantly enhancing retinal structure segmentation and disease classification. Our OCTA dataset (the first dataset in this research area with paired $3\times 3$ and $6\times \text{6}\,\text{mm}^{2}$ OCTA images) and code are publicly available.
Huaying Hao, Shaoyi Leng, Yanda Meng, Yonghuai Liu, Yalin Zheng, Huazhu Fu, Jiong Zhang 0004, Quanyong Yi, Yue Liu 0005, Jingfeng Zhang, Yitian Zhao
IEEE J. Biomed. Health Informatics11
2026 Enhancing Trustworthiness of Semantic Segmentation in Cataract Surgery Videos via Intra-Phase Label Propagation
abstract
Accurate segmentation of semantic features is a pivotal procedure for cataract surgery assistance, surgical skill assessment and related applications. However, previous studies have failed to consider the instance-level feature similarity of instruments across different surgical phases in cataract surgery videos, leading to unreliable decision-making regarding instrument categories. In this study, we propose a label propagation framework to effectively leverage the consistency of phase-specific instruments, which utilizes the initial frame labels from each surgical phase to predict masks for the remaining frames, achieving precise and trustworthy semantic segmentation of cataract surgery videos. Specifically, we design a pseudo-label generation and filtering strategy to automatically obtain highly reliable initial frame labels for each surgical phase. In addition, we establish a fixed-size memory bank with an adaptive update module to ensure long-term applicability in real surgical environments. To address the common problem of blurred edges in cataract surgery scenes, we develop a semantic edge perception module to allow the model to focus on and distinguish the edges of different objects. The proposed method achieved an mIoU of 80.7% and 88.8% on a publicly available dataset (14 categories) and a private dataset (12 categories) with a total of 9,723 frames, respectively, significantly outperforming the state-of-the-art methods and other label propagation-based approaches. Furthermore, our method minimizes memory consumption and maintains about 30 FPS while processing long video sequences.
Mingen Zhang, Xu Chen 0030, Botian Zheng, Donghan Wu, Jinxian Zhang, Yufei Wu 0013, Yonghuai Liu, Yitian Zhao
IEEE J. Biomed. Health Informatics9
2026 Progressive Distillation for Incremental Learning in Corneal Confocal Microscopy Segmentation
abstract
The morphological changes of corneal structures captured by corneal confocal microscopy (CCM), such as corneal nerves, Langerhans cells, stromal cells, etc., are closely related to various ocular and systemic diseases. Current CCM segmentation methods primarily focus on single-task, which limits their broad applicability in clinical practice. The absence of a standardized benchmark further presents a significant challenge in evaluating new methods. To this end, this paper presents a novel incremental learning-based approach for multi-structure segmentation in CCM images and a new benchmark. Specifically, we first propose a data fingerprint distillation (FIND) module to encode task-relevant knowledge by extracting compact representations of structures from CCM images via structural importance mapping. Building on FIND, we propose a progressive task-guided adapter learning (ProTA) strategy, which refines the model's representation of structures through a series of "easy-to-hard" distillation stages. ProTA dynamically adjusts the scope of task-relevant knowledge extracted by FIND, thereby improving the model's ability to accurately discriminate between multiple structures while enhancing knowledge transfer efficiency. Extensive experiments demonstrate that the proposed method achieves the state-of-the-art performance in terms of all corneal structures segmentation. We also demonstrate our approach's plug-and-play capability across four other medical image modalities, suggesting its potential as a general incremental learning tool. Additionally, this work seeks to provide a benchmark tool comprising a comprehensive dataset and their fine manual annotation, as well as unified benchmarking evaluations for state-of-the-art methods. All the dataset, source code and evaluation tool are publicly available at https://github.com/iMED-Lab/CCM-Pro.
Hongshuo Li, Baikai Ma, Lei Mou, Yonghuai Liu, Qinxiang Zheng, Yitian Zhao
IEEE Trans. Medical Imaging7
2026 Beyond Correlation: Causal Intervention for Multi-Label Medical Image Diagnosis
abstract
This paper addresses the challenge of multi-disease diagnosis by integrating causal reasoning into the diagnostic framework. In clinical practice, multiple conditions often co-occur, making multi-disease diagnosis more relevant than isolated single-disease cases. However, most deep learning methods focus on single-disease detection and fail to capture the complexity of diagnosing concurrent conditions. Even in multi-label settings, existing approaches mainly rely on correlation-based inference, capturing statistical associations rather than true causal relationships. This can lead to spurious feature-disease associations, where features linked to one disease are mistakenly attributed to another due to frequent co-occurrence, ultimately undermines diagnostic accuracy and interpretability. To address this challenge, we propose a novel framework that incorporates causal intervention into multi-label medical image diagnosis, enabling the model to identify true causal signals rather than misleading correlations arising from co-occurring diseases. Specifically, we model latent disease-related confounders and apply backdoor adjustment to disentangle genuine causal effects from spurious associations. This is achieved by implicitly learning shared feature representations that serve as confounding variables, which are then used to refine image-derived features during prediction. The resulting causal adjustment allows the model to focus on disease-specific cues, improving accuracy and interpretability. Extensive experiments on four diverse medical imaging datasets: ODIR (color fundus photography), LID-FFA (fundus fluorescein angiography), Endo (colonoscopy), and Chestpert (X-ray) demonstrate that our method consistently outperforms existing approaches. Furthermore, our model also effectively separates the diagnosis of co-occurring diseases, highlighting the potential of causal reasoning to enhance the reliability and clinical applicability of AI-assisted diagnosis. The source code is publicly available at https://github.com/davelailai/BankCausal.git.
Jianyang Xie, Yitian Zhao, Xiuju Chen, Yanda Meng, He Zhao 0002, Uazman Alam, Yalin Zheng
IEEE Trans. Medical Imaging2
2026 Predictive Reasoning With Augmented Anomaly Contrastive Learning for Compositional Visual Relations
abstract
While visual reasoning for simple analogies has received significant attention, compositional visual relations (CVR) remain relatively unexplored due to their greater complexity. To solve CVR tasks, we propose Predictive Reasoning with Augmented Anomaly Contrastive Learning (PR-A$^{2}$CL), i.e., to identify an outlier image given three other images that follow the same compositional rules. To address the challenge of modelling abundant compositional rules, an Augmented Anomaly Contrastive Learning is designed to distil discriminative and generalizable features by maximizing similarity among normal instances while minimizing similarity between normal and anomalous outliers. More importantly, a predict-and-verify paradigm is introduced for rule-based reasoning, in which a series of Predictive Anomaly Reasoning Blocks (PARBs) iteratively leverage features from three out of the four images to predict those of the remaining one. Throughout the subsequent verification stage, the PARBs progressively pinpoint the specific discrepancies attributable to the underlying rules. Experimental results on SVRT, CVR and MC$^{2}$R datasets show that PR-A$^{2}$CL significantly outperforms state-of-the-art reasoning models.
Chengtai Li, Yuting He 0002, Jianfeng Ren, Ruibin Bai, Yitian Zhao, Heng Yu 0001, Xudong Jiang 0001
IEEE Trans. Multim.5
2025 DARR: A Dual-Branch Arithmetic Regression Reasoning Framework for Solving Machine Number Reasoning
abstract
Abstract visual reasoning (AVR) is a critical ability of humans, and it has been widely studied, but arithmetic visual reasoning, a unique task in AVR to reason over number sense, is less studied in the literature. To facilitate this research, we construct a Machine Number Reasoning (MNR) dataset to assess the model's ability in arithmetic visual reasoning over number sense and spatial layouts. To solve the MNR tasks, we propose a Dual-branch Arithmetic Regression Reasoning (DARR) framework, which includes an Intra-Image Arithmetic Regression Reasoning (IIARR) module and a Cross-Image Arithmetic Regression Reasoning (CIARR) module. The IIARR includes a set of Intra-Image Regression Blocks to identify the correct number orders and the underlying arithmetic rules within individual images, and an Order Gate to determine the correct number order. The CIARR establishes the arithmetic relations across different images through a `3-to-1' regressor and a set of `2-to-1' regressors, with a Selection Gate to select the most suitable `2-to-1' regressor and a gated fusion to combine the two kinds of regressors. Experiments on the MNR dataset show that the DARR outperforms state-of-the-art models for arithmetic visual reasoning.
Chengtai Li, Yee Yang Tan, Yuting He 0002, Jianfeng Ren, Ruibin Bai, Yitian Zhao, Heng Yu 0001, Xudong Jiang 0001
AAAI6
2025 Frequency - Semantic Dual - Driven Segmentation of Peripapillary Atrophy in Retinal Fundus Images
abstract
Peripapillary atrophy (PPA) is a critical imaging biomarker for the diagnosis and grading of pathological myopia (PM), and its accurate segmentation plays an essential role in clinical evaluation and longitudinal monitoring. However, automatic PPA segmentation from color fundus images remains challenging due to blurred and discontinuous boundaries, low tissue contrast, and noise interference, which limit the performance of existing methods. In this study, we propose a novel segmentation framework, termed FSD-PPA (Frequency-Semantic Dual-Driven Segmentation Network), to address these limitations. The proposed FSD-PPA integrates a high-frequency edge extraction module to enhance anatomical boundary perception, a low-frequency compensation module to restore tissue structural integrity, and a multi-scale semantic enhancement module to improve cross-level contextual understanding. Experimental results on an expert-annotated PPA dataset and the public PALM dataset demonstrate that our method achieves Dice scores of 83.48 % and 84.30 %, respectively, outperforming existing approaches. These results confirm two major advantages of FSD-PPA: (1) high segmentation accuracy, enabling precise identification of complex PPA morphology; and (2) strong generalization ability, with stable performance across different datasets. This study provides a robust and reliable solution for automatic PPA segmentation. Clinically, it holds potential as an assistive tool for early PM screening and offers an objective imaging basis for quantitative assessment and monitoring of myopic progression, thereby facilitating intelligent diagnosis of myopia-associated fundus lesions.lligent diagnosis systems for myopia-related fundus lesions.
Wanbo Wang, Hongshuo Li, Leilei Yuan, Yaxuan Zhao, Shaodong Ma, Yitian Zhao
BIBM7
2025 AdaptSAM: Adaptive SAM for Cross-Domain Few-Shot Medical Image Segmentation
abstract
Deep networks excel in medical image segmentation with large annotated datasets but struggle with generalization to unseen out-of-domain data, a common challenge in clinical settings. Although the Segment Anything Model (SAM) demonstrates strong prompt-driven generalization in natural image segmentation, its application to clinical segmentation, especially in cross-domain and few-shot scenarios, faces limitations due to insufficient domain-specific features, weak structural fidelity, and prompt instability. To address these challenges, we propose AdaptSAM, an innovative framework to improve cross-domain few-shot medical image segmentation through three key components: Frequency-aware, Semantically-aligned, and Promptadaptive strategies. The Frequency-domain Multi-scale Feature Enhancement (FFE) module extracts frequency-aware features and enriches them with context-sensitive semantics, mitigating domain-specific feature loss. The enhanced features are fused with the Hierarchical Semantic Refinement (HSR) module, utilizing high-level semantic activations to recalibrate shallow-layer features, improving fine-structure fidelity and boundary preservation. Additionally, the Dynamic Curriculum Prompt (DCP) mechanism adjusts prompt box sizes during training, guiding the model to learn object-boundary interactions and background context in a coarse-to-fine manner, thereby aligning prompts with target domain features and enhancing segmentation robustness across domain shifts. AdaptSAM outperforms state-of-the-art methods on ten public medical segmentation datasets, achieving 6.22 % higher Dice than SAM-based methods and 10.37 % higher than fully supervised domain adaptation, showcasing superior cross-domain generalization with just one labeled sample. Code is provided at: https://github.com/Ggllllllll/AdaptSAM.
Wei Zhou 0021, Guilin Guan, Qifeng Yan, Yitian Zhao
BIBM4
2025 Are Spatial-Temporal Graph Convolution Networks for Human Action Recognition Over-Parameterized?
abstract
Spatial-temporal graph convolutional networks (ST-GCNs) showcase impressive performance in skeleton-based human action recognition (HAR). However, despite the development of numerous models, their recognition performance does not differ significantly after aligning the input settings. With this observation, we hypothesize that ST-GCNs are over-parameterized for HAR, a conjecture subsequently confirmed through experiments employing the lottery ticket hypothesis. Additionally, a novel sparse ST-GCNs generator is proposed, which trains a sparse architecture from a randomly initialized dense network while maintaining comparable performance levels to the dense components. Moreover, we generate multi-level sparsity ST-GCNs by integrating sparse structures at various sparsity levels and demonstrate that the assembled model yields a significant enhancement in HAR performance. Thorough experiments on four datasets, including NTU-RGB+D 60(120), Kinetics-400, and FineGYM, demonstrate that the proposed sparse ST-GCNs can achieve comparable performance to their dense components. Even with 95% fewer parameters, the sparse ST-GCNs exhibit a degradation of1% in top-1 accuracy. The code is available at https://github.com/davelailai/Sparse-ST-GCN.
Jianyang Xie, Yitian Zhao, Yanda Meng, He Zhao 0002, Anh Nguyen 0003, Yalin Zheng
CVPR2
2025 DBCR: Exploiting Both Intra-cluster and Extra-cluster Relations for Compositional Reasoning
abstract
Most existing models for abstract visual reasoning perform poorly in compositional visual reasoning (CVR), due to complex nature of compositional rules and difficulties in distinguishing tiny rule differences between outliers and normal images. To tackle the challenges, we propose a Dual-Branch Compositional Reasoning (DBCR) model, exploiting both intra-cluster relations among the cluster of normal images and extra-cluster relations between normal images and outliers. Specifically, we design one branch of Intra-Cluster Regression Reasoning Blocks (ICR2Bs) to encapsulate common relations among normal images through hierarchical regressing reasoning, and the other branch of Contrastive Attention Reasoning Blocks (CARBs) to exploit extra-cluster differences between normal images and outliers through self-attention. Simultaneously minimizing the regression errors in ICR2Bs and maximizing the extra-cluster differences in CARBs help identify the correct cluster of normal images. Experimental results on two CVR datasets show that the proposed DBCR consistently outperforms state-of-the-art models. The code is available at https://github.com/He1mont/DBCR.
Chengtai Li, Guosheng Su, Jianfeng Ren, Ruibin Bai, Yitian Zhao, Xudong Jiang 0001
ICASSP5
2025 MorphoBoost: Morphology-Driven Boundary Enhancement Model for Accurate Segmentation of Langerhans Cells in Corneal Confocal Microscopy Images
Hongshuo Li, Ankai Dong, Tiande Zhang, Shijia Zhou, Yalin Zheng, Lei Mou, Yitian Zhao
MICCAI (13)7
2025 A Frequency-Aware Self-supervised Learning for Ultra-Wide-Field Image Enhancement
Weicheng Liao, Jianyang Xie, Yalin Zheng, Yuhui Ma, Yitian Zhao
MICCAI (13)6
2025 Fairness-Aware vCDR-Controlled Generation for Glaucoma Diagnosis
Shuran Yang, Feixiang Zhou, Meng Wang 0038, Yitian Zhao, Yalin Zheng, Yanda Meng
MICCAI (9)9
2025 Robust Incomplete-Modality Alignment for Ophthalmic Disease Grading and Diagnosis via Labeled Optimal Transport
Qinkai Yu, Jianyang Xie, Yitian Zhao, Cheng Chen 0013, Jun Cheng 0003, Lu Liu 0001, Yalin Zheng, Yanda Meng
MICCAI (15)3
2025 Parameterized Diffusion Optimization Enabled Autoregressive Ordinal Regression for Diabetic Retinopathy Grading
Qinkai Yu, Wei Zhou 0021, Hantao Liu, Yanyu Xu 0001, Meng Wang 0038, Yitian Zhao, Huazhu Fu, Xujiong Ye, Yalin Zheng, Yanda Meng
MICCAI (15)6
2025 GLCP: Global-to-Local Connectivity Preservation for Tubular Structure Segmentation
Feixiang Zhou, Zhuangzhi Gao, He Zhao 0002, Jianyang Xie, Yanda Meng, Yitian Zhao, Gregory Yoke Hong Lip, Yalin Zheng
MICCAI (16)6
2025 DSRF: A Dynamic and Scalable Reasoning Framework for Solving RPMs
abstract
Abstract Visual Reasoning (AVR) entails discerning latent patterns in visual data and inferring underlying rules. Existing solutions often lack scalability and adaptability, as deep architectures tend to overfit training data, and static neural networks fail to dynamically capture diverse rules. To tackle the challenges, we propose a Dynamic and Scalable Reasoning Framework (DSRF) that greatly enhances the reasoning ability by widening the network instead of deepening it, and dynamically adjusting the reasoning network to better fit novel samples instead of a static network. Specifically, we design a Multi-View Reasoning Pyramid (MVRP) to capture complex rules through layered reasoning to focus features at each view on distinct combinations of attributes, widening the reasoning network to cover more attribute combinations analogous to complex reasoning rules. Additionally, we propose a Dynamic Domain-Contrast Prediction (DDCP) block to handle varying task-specific relationships dynamically by introducing a Gram matrix to model feature distributions, and a gate matrix to capture subtle domain differences between context and target features. Extensive experiments on six AVR tasks demonstrate DSRF’s superior performance, achieving state-of-the-art results under various settings. Code is available here: https://github.com/UNNCRoxLi/DSRF.
Chengtai Li, Yuting He 0002, Jianfeng Ren, Ruibin Bai, Yitian Zhao, Xudong Jiang 0001
NeurIPS5
2025 Keypoints-based heterogeneous graph convolutional networks for construction machinery activity classification
abstract
Existing computer vision-based approaches struggle to identify machinery actions due to the challenges posed by environmental complexity and various obstructions in construction. This study introduces a novel two-stage framework that benefits from a newly proposed Residual Fusion Graph Convolution Network (RFGCN) to classify machinery actions with enhanced robustness and accuracy. The framework first extracts key machinery components from video data, subsequently transforming them into a graph-based representation. This spatio-temporal graph is then fed into the RFGCN model, specifically designed to overcome issues like partial obstructions and missing information common in busy construction sites. Experimental evaluations reveal the method’s high efficacy, achieving an accuracy of up to 96.4% and outperforming state-of-the-art. Additionally, the proposed RFGCN model achieved state-of-the-art performance on four established benchmark datasets for graph classification using spatial data only. These results suggest the potential of the proposed framework in facilitating the transition towards more intelligent and automated construction sites.
Shuozhi Wang, Yitian Zhao, Yifan Zhao 0001
Expert Syst. Appl.2
2025 Interweaving Insights: High-Order Feature Interaction for Fine-Grained Visual Recognition
abstract
This paper presents a novel approach for Fine-Grained Visual Classification (FGVC) by exploring Graph Neural Networks (GNNs) to facilitate high-order feature interactions, with a specific focus on constructing both inter- and intra-region graphs. Unlike previous FGVC techniques that often isolate global and local features, our method combines both features seamlessly during learning via graphs. Inter-region graphs capture long-range dependencies to recognize global patterns, while intra-region graphs delve into finer details within specific regions of an object by exploring high-dimensional convolutional features. A key innovation is the use of shared GNNs with an attention mechanism coupled with the Approximate Personalized Propagation of Neural Predictions (APPNP) message-passing algorithm, enhancing information propagation efficiency for better discriminability and simplifying the model architecture for computational efficiency. Additionally, the introduction of residual connections improves performance and training stability. Comprehensive experiments showcase state-of-the-art results on benchmark FGVC datasets, affirming the efficacy of our approach. This work underscores the potential of GNN in modeling high-level feature interactions, distinguishing it from previous FGVC methods that typically focus on singular aspects of feature representation. Our source code is available at https://github.com/Arindam-1991/I2-HOFI.
Arindam Sikdar, Yonghuai Liu, Siddhardha Kedarisetty, Yitian Zhao, Amr Ahmed 0002, Ardhendu Behera
Int. J. Comput. Vis.4
2025 Beyond the eye: A relational model for early dementia detection using retinal OCTA images
Shouyue Liu, Jinkui Hao, Yonghuai Liu, Huazhu Fu, Yitian Zhao
Medical Image Anal.10
2025 Neovascularization Segmentation via a Multilateral Interaction-Enhanced Graph Convolutional Network
abstract
Choroidal neovascularization (CNV), a primary characteristic of wet age-related macular degeneration (wet AMD), represents a leading cause of blindness worldwide. In clinical practice, optical coherence tomography angiography (OCTA) is commonly used for studying CNV-related pathological changes, due to its micron-level resolution and non-invasive nature. Thus, accurate segmentation of CNV regions and vessels in OCTA images is crucial for clinical assessment of wet AMD. However, challenges existed due to irregular CNV shapes and imaging limitations like projection artifacts, noises and boundary blurring. Moreover, the lack of publicly available datasets constraints the CNV analysis. To address these challenges, this paper constructs the first publicly accessible CNV dataset (CNVSeg), and proposes a novel multilateral graph convolutional interaction-enhanced CNV segmentation network (MTG-Net). This network integrates both region and vessel morphological information, exploring semantic and geometric duality constraints within the graph domain. Specifically, MTG-Net consists of a multi-task framework and two graph-based cross-task modules: Multilateral Interaction Graph Reasoning (MIGR) and Multilateral Reinforcement Graph Reasoning (MRGR). The multi-task framework encodes rich geometric features of lesion shapes and surfaces, decoupling the image into three task-specific feature maps. MIGR and MRGR iteratively reason about higher-order relationships across tasks through a graph mechanism, enabling complementary optimization for task-specific objectives. Additionally, an uncertainty-weighted loss is proposed to mitigate the impact of artifacts and noise on segmentation accuracy. Experimental results demonstrate that MTG-Net outperforms existing methods, achieving a Dice socre of 87.21% for region segmentation and 88.12% for vessel segmentation.
Tao Chen 0003, Dan Zhang 0026, Da Chen 0002, Huazhu Fu, Shanshan Wang 0002, Laurent D. Cohen, Yitian Zhao, Quanyong Yi, Jiong Zhang 0004
IEEE Trans. Pattern Anal. Mach. Intell.8
2025 3D microvascular reconstruction in retinal OCT angiography images via domain-adaptive learning
Jiong Zhang 0004, Yonghuai Liu, Dan Zhang 0026, Jianyang Xie, Tao Chen 0003, Yalin Zheng, Huazhu Fu, Yitian Zhao
Pattern Recognit.9
2025 MIFNet: Learning Modality-Invariant Features for Generalizable Multimodal Image Matching
abstract
Many keypoint detection and description methods have been proposed for image matching or registration. While these methods demonstrate promising performance for single-modality image matching, they often struggle with multimodal data because the descriptors trained on single-modality data tend to lack robustness against the non-linear variations present in multimodal data. Extending such methods to multimodal image matching often requires well-aligned multimodal data to learn modality-invariant descriptors. However, acquiring such data is often costly and impractical in many real-world scenarios. To address this challenge, we propose a modality-invariant feature learning network (MIFNet) to compute modality-invariant features for keypoint descriptions in multimodal image matching using only single-modality training data. Specifically, we propose a novel latent feature aggregation module and a cumulative hybrid aggregation module to enhance the base keypoint descriptors trained on single-modality data by leveraging pre-trained features from Stable Diffusion models. We validate our method with recent keypoint detection and description methods in three multimodal retinal image datasets (CF-FA, CF-OCT, EMA-OCTA) and two remote sensing datasets (Optical-SAR and Optical-NIR). Extensive experiments demonstrate that the proposed MIFNet is able to learn modality-invariant feature for multimodal image matching without accessing the targeted modality and has good zero-shot generalization ability. The code will be released at https://github.com/lyp-deeplearning/MIFNet.
Yepeng Liu 0002, Zhichao Sun 0004, Baosheng Yu, Yitian Zhao, Bo Du 0001, Yongchao Xu, Jun Cheng 0003
IEEE Trans. Image Process.4
2025 Rethinking Data Augmentation for Single-Source Domain Generalization in OCT Image Segmentation
abstract
Domain shifts between samples acquired with different instruments are one of the major challenges in accurate segmentation of Optical Coherence Tomography (OCT) images. Given that OCT images may be acquired with different devices in different clinical centers, this study presents astyle and structure data augmentation (SSDA) method to improve the adaptability of segmentation models. Inspired by our initial analysis of OCT domain differences, we propose an innovative hypothesis that domain shifts are primarily due to differences in image style and anatomical structure, which further guides the design of our method. By designing a modality-specific NURBS curve for style enhancement and implementing global and local elastic deformation fields, SSDA addresses both stylistic and structural variations in OCT data. Global deformations simulate changes in retinal curvature, while local deformations model layer-specific changes observed in OCT images. We validate our hypothesis through a comprehensive evaluation conducted on five OCT data domains, each differing in device type and imaging conditions. We train models on each of these domains for single-domain generalisation experiments and evaluate performance on the remaining unseen domains. The results show that SSDA outperforms existing methods when segmenting OCT images from different sources with different requirements for retinal layer segmentation. Specifically, across five different source domain generalisation experiments, SSDA achieves approximately 1.6% higher Dice and 2.6% improved MIOU, underscoring its superior segmentation accuracy and robust generalisation across all evaluated unseen domains.
Shaodong Ma, Yonghuai Liu, Yuhui Ma, Lei Mou, Yitian Zhao
IEEE J. Biomed. Health Informatics7
2025 $\text{MR}^{2}$-Net: Retinal OCTA Image Stitching via Multi-Scale Representation Learning and Dynamic Location Guidance
abstract
Optical coherence tomography angiography (OCTA) plays a crucial role in quantifying and analyzing retinal vascular diseases. However, the limited field of view (FOV) inherent in most commercial OCTA imaging systems poses a significant challenge for clinicians, restricting the possibility to analyze larger retinal regions of high resolution. Automatic stitching of OCTA scans in adjacent regions may provide a promising solution to extend the region of interest. However, commonly-used stitching algorithms face difficulties in achieving effective alignment due to noise, artifacts and dense vasculature present in OCTA images. To address these challenges, we propose a novel retinal OCTA image stitching network, named -Net, which integrates multi-scale representation learning and dynamic location guidance. In the first stage, an image registration network with a progressive multi-resolution feature fusion is proposed to derive deep semantic information effectively. Additionally, we introduce a dynamic guidance strategy to locate the foveal avascular zone (FAZ) and constrain registration errors in overlapping vascular regions. In the second stage, an image fusion network based on multiple mask constraints and adjacent image aggregation (AIA) strategies is developed to further eliminate the artifacts in the overlapping areas of stitched images, thereby achieving precise vessel alignment. To validate the effectiveness of our method, we conduct a series of experiments on two delicately constructed datasets, i.e., OPTOVUE-OCTA and SVision-OCTA. Experimental results demonstrate that our method outperforms other image stitching methods and effectively generates high-quality wide-field OCTA images, achieving a structural similarity index (SSIM) score of 0.8264 and 0.8014 on the two datasets, respectively.
Haiting Mao, Yuhui Ma, Dan Zhang 0026, Yanda Meng, Shaodong Ma, Yuchuan Qiao, Huazhu Fu, Caifeng Shan, Da Chen 0002, Yitian Zhao, Jiong Zhang 0004
IEEE J. Biomed. Health Informatics10
2025 Randomness-Restricted Diffusion Model for Ocular Surface Structure Segmentation
abstract
Ocular surface diseases affect a significant portion of the population worldwide. Accurate segmentation and quantification of different ocular surface structures are crucial for the understanding of these diseases and clinical decision-making. However, the automated segmentation of the ocular surface structure is relatively unexplored and faces several challenges. Ocular surface structure boundaries are often inconspicuous and obscured by glare from reflections. In addition, the segmentation of different ocular structures always requires training of multiple individual models. Thus, developing a one-model-fits-all segmentation approach is desirable. In this paper, we introduce a randomness-restricted diffusion model for multiple ocular surface structure segmentation. First, a time-controlled fusion-attention module (TFM) is proposed to dynamically adjust the information flow within the diffusion model, based on the temporal relationships between the network's input and time. TFM enables the network to effectively utilize image features to constrain the randomness of the generation process. We further propose a low-frequency consistency filter and a new loss to alleviate model uncertainty and error accumulation caused by the multi-step denoising process. Extensive experiments have shown that our approach can segment seven different ocular surface structures. Our method performs better than both dedicated ocular surface segmentation methods and general medical image segmentation methods. We further validated the proposed method over two clinical datasets, and the results demonstrated that it is beneficial to clinical applications, such as the meibomian gland dysfunction grading and aqueous deficient dry eye diagnosis.
Huaying Hao, Yifan Zhao 0001, Yanda Meng, Jiang Liu 0001, Yalin Zheng, Wei Chen 0089, Yitian Zhao
IEEE Trans. Medical Imaging9
2025 RSAPower: Random Style Augmentation Driven Structure Perception Network for Generalized Retinal OCT Fluid Segmentation
abstract
Optical Coherence Tomography (OCT) imaging is extensively utilized for non-invasive observation of pathological conditions, such as retinal fluid-associated diseases. Accurate fluid segmentation in OCT images is therefore critical for quantifying disease severity and aiding clinical decision-making. However, achieving precise segmentation remains challenging due to pathological variations in shape and size, uncertain boundaries, and low contrast of fluid. Most importantly, variability in OCT image styles across different vendors and centers significantly affects fluid segmentation, leading to poor generalization to unseen domains. To address this, we propose a novel method, RSAPower, to enhance the generalization ability of fluid perception networks via style augmentation for retinal fluid segmentation. Specifically, RSAPower comprises a plug-and-play random style transform augmentation (RSTAug) module and a novel fluid perception network (FLPNet) for end-to-end training. The RSTAug module generates new random-style data from the source domain, preserving realistic pathological and structural features. The FLPNet benefits from a novel hybrid structure attention (HSA) module to perceive fluid's spatial features and long-range dependence. Furthermore, FLPNet adapts to the diverse augmented data through a saliency-guided multi-scale attention (SGMA) block, boosting its segmentation performance. We validate RSAPower against various state-of-the-art methods using two publicly available datasets, Retouch and Kermany. Experimental results demonstrate the proposed method's superior generalization ability and effectiveness in fluid segmentation.
Chenggang Lu, Zhitao Guo, Dan Zhang 0026, Lei Mou, Jinli Yuan, Shaodong Ma, Da Chen 0002, Yitian Zhao, Kewen Xia, Jiong Zhang 0004
IEEE Trans. Medical Imaging8
2025 DSCA: A Digital Subtraction Angiography Sequence Dataset and Spatio-Temporal Model for Cerebral Artery Segmentation
abstract
Cerebrovascular diseases (CVDs) remain a leading cause of global disability and mortality. Digital Subtraction Angiography (DSA) sequences, recognized as the gold standard for diagnosing CVDs, can clearly visualize the dynamic flow and reveal pathological conditions within the cerebrovasculature. Therefore, precise segmentation of cerebral arteries (CAs) and classification between their main trunks and branches are crucial for physicians to accurately quantify diseases. However, achieving accurate CA segmentation in DSA sequences remains a challenging task due to small vessels with low contrast, and ambiguity between vessels and residual skull structures. Moreover, the lack of publicly available datasets limits exploration in the field. In this paper, we introduce a DSA Sequence-based Cerebral Artery segmentation dataset (DSCA), the publicly accessible dataset designed specifically for pixel-level semantic segmentation of CAs. Additionally, we propose DSANet, a spatio-temporal network for CA segmentation in DSA sequences. Unlike existing DSA segmentation methods that focus only on a single frame, the proposed DSANet introduces a separate temporal encoding branch to capture dynamic vessel details across multiple frames. To enhance small vessel segmentation and improve vessel connectivity, we design a novel TemporalFormer module to capture global context and correlations among sequential frames. Furthermore, we develop a Spatio-Temporal Fusion (STF) module to effectively integrate spatial and temporal features from the encoder. Extensive experiments demonstrate that DSANet outperforms other state-of-the-art methods in CA segmentation, achieving a Dice of 0.9033.
Jiong Zhang 0004, Qihang Xie, Lei Mou, Dan Zhang 0026, Da Chen 0002, Caifeng Shan, Yitian Zhao, Ruisheng Su, Mengguo Guo
IEEE Trans. Medical Imaging7
2024 Scale Optimization Using Evolutionary Reinforcement Learning for Object Detection on Drone Imagery
abstract
Object detection in aerial imagery presents a significant challenge due to large scale variations among objects. This paper proposes an evolutionary reinforcement learning agent, integrated within a coarse-to-fine object detection framework, to optimize the scale for more effective detection of objects in such images. Specifically, a set of patches potentially containing objects are first generated. A set of rewards measuring the localization accuracy, the accuracy of predicted labels, and the scale consistency among nearby patches are designed in the agent to guide the scale optimization. The proposed scale-consistency reward ensures similar scales for neighboring objects of the same category. Furthermore, a spatial-semantic attention mechanism is designed to exploit the spatial semantic relations between patches. The agent employs the proximal policy optimization strategy in conjunction with the evolutionary strategy, effectively utilizing both the current patch status and historical experience embedded in the agent. The proposed model is compared with state-of-the-art methods on two benchmark datasets for object detection on drone imagery. It significantly outperforms all the compared methods. Code is available at https://github.com/UNNC-CV/EvOD/.
Jialu Zhang 0003, Jianfeng Ren, Qian Zhang 0018, Yitian Zhao, Ruibin Bai, Xiangjian He, Jiang Liu 0001
AAAI6
2024 Dynamic Semantic-Based Spatial Graph Convolution Network for Skeleton-Based Human Action Recognition
abstract
Graph convolutional networks (GCNs) have attracted great attention and achieved remarkable performance in skeleton-based action recognition. However, most of the previous works are designed to refine skeleton topology without considering the types of different joints and edges, making them infeasible to represent the semantic information. In this paper, we proposed a dynamic semantic-based graph convolution network (DS-GCN) for skeleton-based human action recognition, where the joints and edge types were encoded in the skeleton topology in an implicit way. Specifically, two semantic modules, the joints type-aware adaptive topology and the edge type-aware adaptive topology, were proposed. Combining proposed semantics modules with temporal convolution, a powerful framework named DS-GCN was developed for skeleton-based action recognition. Extensive experiments in two datasets, NTU-RGB+D and Kinetics-400 show that the proposed semantic modules were generalized enough to be utilized in various backbones for boosting recognition accuracy. Meanwhile, the proposed DS-GCN notably outperformed state-of-the-art methods. The code is released here https://github.com/davelailai/DS-GCN
Jianyang Xie, Yanda Meng, Yitian Zhao, Anh Nguyen 0003, Xiaoyun Yang, Yalin Zheng
AAAI3
2024 Clinical Insight-Augmented Multi-View Learning for Alzheimer's Detection in Retinal OCTA Images
abstract
Alzheimer’s disease (AD) poses a significant global challenge, with a notable absence of accessible and cost-effective diagnostic tools for widespread AD detection. The retina, mirroring the brain in anatomy and physiology, has emerged as a potential avenue for rapid AD identification through retinal imaging. The current retinal image-based AD detection methods usually focus primarily on the macular area, but ignore the potential value that the optic disc region may have for the detection task. In this study, we leverage both macular- and disc-centered OCTA images and propose a multi-region fusion framework for AD detection. Based on clinical evidence, we integrate handcrafted features into the framework to improve model performance and interpretability. Specifically, vascular morphological parameters extracted from the macular and disc regions are used as input to a revalued KNN model to improve predictive capabilities. Furthermore, recognizing the significance of extracting and utilizing complementary information from the macular and optic disc regions, we propose an uncertainty-guided strategy based on Dempster-Shefer Theory (DST) to fuse knowledge from different regions. This approach considers each region’s forecast quality and significantly improves the effectiveness and robustness of the model. Through comparative analysis with existing methods, we have demonstrated that our method outperforms the state-of-the-art ones and provides more valuable pathological evidence for the association between retinal vascular changes and AD.
Yuandi Zhang, Jinkui Hao, Botian Zheng, Yonghuai Liu, Yanda Meng, Jiong Zhang 0004, Yang Chen 0008, Yitian Zhao
BIBM9
2024 Regression Residual Reasoning with Pseudo-labeled Contrastive Learning for Uncovering Multiple Complex Compositional Relations
Chengtai Li, Yuting He 0002, Jianfeng Ren, Ruibin Bai, Yitian Zhao, Heng Yu 0001, Xudong Jiang 0001
IJCAI5
2024 Open-Set Semi-supervised Medical Image Classification with Learnable Prototypes and Outlier Filter
Along He, Tao Li 0022, Yitian Zhao, Junyong Zhao, Huazhu Fu
MICCAI (11)3
2024 MPMNet: Modal Prior Mutual-Support Network for Age-Related Macular Degeneration Classification
Huaying Hao, Dan Zhang 0026, Huazhu Fu, Caifeng Shan, Yitian Zhao, Jiong Zhang 0004
MICCAI (1)7
2024 A Hyperreflective Foci Segmentation Network for OCT Images with Multi-dimensional Semantic Enhancement
Xingguo Wang, Yuhui Ma, Yalin Zheng, Jiong Zhang 0004, Yonghuai Liu, Yitian Zhao
MICCAI (1)7
2024 Multi-disease Detection in Retinal Images Guided by Disease Causal Estimation
Jianyang Xie, Xiuju Chen, Yitian Zhao, Yanda Meng, He Zhao 0002, Anh Nguyen 0003, Yalin Zheng
MICCAI (1)3
2024 DSNet: A Spatio-Temporal Consistency Network for Cerebrovascular Segmentation in Digital Subtraction Angiography Sequences
Qihang Xie, Dan Zhang 0026, Lei Mou, Shanshan Wang 0002, Yitian Zhao, Mengguo Guo, Jiong Zhang 0004
MICCAI (8)5
2024 CLIP-DR: Textual Knowledge-Guided Diabetic Retinopathy Grading with Ranking-Aware Prompting
Qinkai Yu, Jianyang Xie, Anh Nguyen 0003, He Zhao 0002, Jiong Zhang 0004, Huazhu Fu, Yitian Zhao, Yalin Zheng, Yanda Meng
MICCAI (1)7
2024 Multi-granularity learning of explicit geometric constraint and contrast for label-efficient medical image segmentation and differentiable clinical function assessment
abstract
Automated segmentation is a challenging task in medical image analysis that usually requires a large amount of manually labeled data. However, most current supervised learning based algorithms suffer from insufficient manual annotations, posing a significant difficulty for accurate and robust segmentation. In addition, most current semi-supervised methods lack explicit representations of geometric structure and semantic information, restricting segmentation accuracy. In this work, we propose a hybrid framework to learn polygon vertices, region masks, and their boundaries in a weakly/semi-supervised manner that significantly advances geometric and semantic representations. Firstly, we propose multi-granularity learning of explicit geometric structure constraints via polygon vertices (PolyV) and pixel-wise region (PixelR) segmentation masks in a semi-supervised manner. Secondly, we propose eliminating boundary ambiguity by using an explicit contrastive objective to learn a discriminative feature space of boundary contours at the pixel level with limited annotations. Thirdly, we exploit the task-specific clinical domain knowledge to differentiate the clinical function assessment end-to-end. The ground truth of clinical function assessment, on the other hand, can serve as auxiliary weak supervision for PolyV and PixelR learning. We evaluate the proposed framework on two tasks, including optic disc (OD) and cup (OC) segmentation along with vertical cup-to-disc ratio (vCDR) estimation in fundus images; left ventricle (LV) segmentation at end-diastolic and end-systolic frames along with ejection fraction (LVEF) estimation in two-dimensional echocardiography images. Experiments on nine large-scale datasets of the two tasks under different label settings demonstrate our model’s superior performance on segmentation and clinical function assessment.
Yanda Meng, Jianyang Xie, Jinming Duan 0001, Martha Joddrell, Savita Madhusudhan, Tunde Peto, Yitian Zhao, Yalin Zheng
Medical Image Anal.8
2024 Dynamic Semantic-Based Spatial-Temporal Graph Convolution Network for Skeleton-Based Human Action Recognition
abstract
Human action recognition is an essential topic in computer vision and image processing. Graph convolutional networks (GCNs) have attracted significant attention and achieved noteworthy performance in skeleton-based human action recognition tasks. However, most of the previous graph-based works are designed to refine skeleton topology without considering the types of different joints and edges and the occurrence order of the frames. Such a limitation makes them insufficient to represent intrinsic semantic information. Differently, we proposed a dynamic semantic-based spatial-temporal graph convolution network (DS-STGCN) to address the challenge. DS-STGCN has two dynamic semantic modules for spatial and temporal contexts respectively. Specifically, the joints and edge types were encoded in the spatial module implicitly, and the occurrence order of frames was encoded in the temporal module implicitly. Extensive experiments on four datasets including NTU-RGB+D 60(120), Kinetics-400, and FineGYM show that our proposed two semantic modules can bring consistent recognition performance improvement with various backbones. Meanwhile, the proposed DS-STGCN notably surpassed state-of-the-art methods on these datasets. Notably, in the more challenging dataset, such as Kinetics-400, our model significantly outperformed other state-of-the-art GCN-based methods by a large margin. The code has been released at https://github.com/davelailai/DS-STGCN.
Jianyang Xie, Yanda Meng, Yitian Zhao, Anh Nguyen 0003, Xiaoyun Yang, Yalin Zheng
IEEE Trans. Image Process.3
2024 COSTA: A Multi-Center TOF-MRA Dataset and a Style Self-Consistency Network for Cerebrovascular Segmentation
abstract
Time-of-flight magnetic resonance angiography (TOF-MRA) is the least invasive and ionizing radiation-free approach for cerebrovascular imaging, but variations in imaging artifacts across different clinical centers and imaging vendors result in inter-site and inter-vendor heterogeneity, making its accurate and robust cerebrovascular segmentation challenging. Moreover, the limited availability and quality of annotated data pose further challenges for segmentation methods to generalize well to unseen datasets. In this paper, we construct the largest and most diverse TOF-MRA dataset (COSTA) from 8 individual imaging centers, with all the volumes manually annotated. Then we propose a novel network for cerebrovascular segmentation, namely CESAR, with the ability to tackle feature granularity and image style heterogeneity issues. Specifically, a coarse-to-fine architecture is implemented to refine cerebrovascular segmentation in an iterative manner. An automatic feature selection module is proposed to selectively fuse global long-range dependencies and local contextual information of cerebrovascular structures. A style self-consistency loss is then introduced to explicitly align diverse styles of TOF-MRA images to a standardized one. Extensive experimental results on the COSTA dataset demonstrate the effectiveness of our CESAR network against state-of-the-art methods. We have made 6 subsets of COSTA with the source code online available, in order to promote relevant research in the community.
Lei Mou, Jinghui Lin, Yifan Zhao 0001, Yonghuai Liu, Shaodong Ma, Jiong Zhang 0004, Wenhao Lv, Tao Zhou 0002, Jiang Liu 0001, Alejandro F. Frangi, Yitian Zhao
IEEE Trans. Medical Imaging11
2023 SPC-Net: Structure-Aware Pixel-Level Contrastive Learning Network for OCTA A/V Segmentation and Differentiation
Huaying Hao, Yuhui Ma, Lijun Guo, Jiong Zhang 0004, Yitian Zhao
CGI (1)6
2023 RBGNet: Reliable Boundary-Guided Segmentation of Choroidal Neovascularization
Tao Chen 0003, Yitian Zhao, Lei Mou, Dan Zhang 0026, Xiayu Xu, Huazhu Fu, Jiong Zhang 0004
MICCAI (4)2
2023 Polar-Net: A Clinical-Friendly Model for Alzheimer's Disease Detection in OCTA Images
Shouyue Liu, Jinkui Hao, Yanwu Xu 0001, Huazhu Fu, Jiang Liu 0001, Yalin Zheng, Yonghuai Liu, Jiong Zhang 0004, Yitian Zhao
MICCAI (7)10
2023 Weakly/Semi-supervised Left Ventricle Segmentation in 2D Echocardiography with Uncertain Region-Aware Contrastive Learning
Yanda Meng, Jianyang Xie, Jinming Duan 0001, Yitian Zhao, Yalin Zheng
PRCV (13)5
2023 Automatic choroid layer segmentation in OCT images via context efficient adaptive network
Qifeng Yan, Jinyu Zhao, Yuhui Ma, Jiang Liu 0001, Jiong Zhang 0004, Yitian Zhao
Appl. Intell.8
2023 Bilateral adaptive graph convolutional network on CT based Covid-19 diagnosis with uncertainty-aware consensus-assisted multiple instance learning
abstract
Coronavirus disease (COVID-19) has caused a worldwide pandemic, putting millions of people's health and lives in jeopardy. Detecting infected patients early on chest computed tomography (CT) is critical in combating COVID-19. Harnessing uncertainty-aware consensus-assisted multiple instance learning (UC-MIL), we propose to diagnose COVID-19 using a new bilateral adaptive graph-based (BA-GCN) model that can use both 2D and 3D discriminative information in 3D CT volumes with arbitrary number of slices. Given the importance of lung segmentation for this task, we have created the largest manual annotation dataset so far with 7,768 slices from COVID-19 patients, and have used it to train a 2D segmentation model to segment the lungs from individual slices and mask the lungs as the regions of interest for the subsequent analyses. We then used the UC-MIL model to estimate the uncertainty of each prediction and the consensus between multiple predictions on each CT slice to automatically select a fixed number of CT slices with reliable predictions for the subsequent model reasoning. Finally, we adaptively constructed a BA-GCN with vertices from different granularity levels (2D and 3D) to aggregate multi-level features for the final diagnosis with the benefits of the graph convolution network's superiority to tackle cross-granularity relationships. Experimental results on three largest COVID-19 CT datasets demonstrated that our model can produce reliable and accurate COVID-19 predictions using CT volumes with any number of slices, which outperforms existing approaches in terms of learning and generalisation ability. To promote reproducible research, we have made the datasets, including the manual annotations and cleaned CT dataset, as well as the implementation code, available at https://doi.org/10.5281/zenodo.6361963.
Yanda Meng, Joshua Bridge, Cliff Addison, Manhui Wang, Cristin Merritt, Stu Franks, Maria Mackey, Steve Messenger, Renrong Sun, Thomas Fitzmaurice, Caroline McCann, Yitian Zhao, Yalin Zheng
Medical Image Anal.13
2023 3D Visual Saliency: An Independent Perceptual Measure or a Derivative of 2D Image Saliency?
abstract
While 3D visual saliency aims to predict regional importance of 3D surfaces in agreement with human visual perception and has been well researched in computer vision and graphics, latest work with eye-tracking experiments shows that state-of-the-art 3D visual saliency methods remain poor at predicting human fixations. Cues emerging prominently from these experiments suggest that 3D visual saliency might associate with 2D image saliency. This paper proposes a framework that combines a Generative Adversarial Network and a Conditional Random Field for learning visual saliency of both a single 3D object and a scene composed of multiple 3D objects with image saliency ground truth to 1) investigate whether 3D visual saliency is an independent perceptual measure or just a derivative of image saliency and 2) provide a weakly supervised method for more accurately predicting 3D visual saliency. Through extensive experiments, we not only demonstrate that our method significantly outperforms the state-of-the-art approaches, but also manage to answer the interesting and worthy question proposed within the title of this paper.
Ran Song 0001, Wei Zhang 0021, Yitian Zhao, Yonghuai Liu, Paul L. Rosin
IEEE Trans. Pattern Anal. Mach. Intell.3
2023 Transportation Object Counting With Graph-Based Adaptive Auxiliary Learning
abstract
This paper proposes an adaptive auxiliary task learning-based approach for transport object counting problems such as humans and vehicles. These problems are essential in many real-world tasks such as video surveillance, traffic monitoring, public security, and urban planning, to aid intelligent transportation systems. Unlike existing auxiliary task learning-based methods, we develop an attention-enhanced adaptively shared backbone network to enable both task-shared and task-tailored features that are learned in an end-to-end manner. The network seamlessly combines a standard Convolution Neural Network (CNN) and a Graph Convolution Network (GCN) for feature extraction and feature reasoning among different domains of tasks. Our approach gains enriched contextual information by iteratively and hierarchically fusing features across different task branches of the adaptive CNN backbone. The whole framework pays special attention to objects’ spatial locations and varied density levels, informed by object (or crowd) segmentation and density level segmentation auxiliary tasks. In particular, thanks to the proposed dilated contrastive density loss function, our network benefits from individual and regional context supervision, along with strengthened robustness. Experiments on six challenging multi-domain datasets demonstrate that our method achieves superior performance compared with state-of-the-art auxiliary task learning-based counting methods. Our code is publicly available.
Yanda Meng, Joshua Bridge, Yitian Zhao, Martha Joddrell, Yihong Qiao, Xiaoyun Yang, Xiaowei Huang 0001, Yalin Zheng
IEEE Trans. Intell. Transp. Syst.3
2023 Guest Editorial Special Issue on Geometric Deep Learning in Medical Imaging
abstract
In recent years, more and more attention has been devoted to geometric deep learning (GDL) and its applications to various problems in medical imaging. Unlike convolutional neural networks (CNNs) limited to 2-D/3-D grid-structured data, GDL can handle non-Euclidean data (i.e., graphs and manifolds) and is hence well-suited for medical imaging data such as structure-function connectivity networks, imaging genetics and omics, spatio-temporal anatomical representations, physics-informed GDL for optimal imaging sampling and acquisition, GDL in imaging inverse problems, etc. However, despite recent advances in GDL research, questions remain on how best to learn representations of non-Euclidean medical imaging data; how to convolve effectively on graphs; how to perform graph pooling/unpooling; how to handle heterogeneous data; and how to improve the interpretability of GDL. After discussing many other domain experts, we identify the need for a special issue that brings to the attention of the medical imaging community these interesting topics.
Huazhu Fu, Yitian Zhao, Pew-Thian Yap, Carola-Bibiane Schönlieb, Alejandro F. Frangi
IEEE Trans. Medical Imaging2
2023 Dual Consistency Enabled Weakly and Semi-Supervised Optic Disc and Cup Segmentation With Dual Adaptive Graph Convolutional Networks
abstract
Glaucoma is a progressive eye disease that results in permanent vision loss, and the vertical cup to disc ratio (vCDR) in colour fundus images is essential in glaucoma screening and assessment. Previous fully supervised convolution neural networks segment the optic disc (OD) and optic cup (OC) from color fundus images and then calculate the vCDR offline. However, they rely on a large set of labeled masks for training, which is expensive and time-consuming to acquire. To address this, we propose a weakly and semi-supervised graph-based network that investigates geometric associations and domain knowledge between segmentation probability maps (PM), modified signed distance function representations (mSDF), and boundary region of interest characteristics (B-ROI) in three aspects. Firstly, we propose a novel Dual Adaptive Graph Convolutional Network (DAGCN) to reason the long-range features of the PM and the mSDF w.r.t. the regional uniformity. Secondly, we propose a dual consistency regularization-based semi-supervised learning paradigm. The regional consistency between the PM and the mSDF, and the marginal consistency between the derived B-ROI from each of them boost the proposed model's performance due to the inherent geometric associations. Thirdly, we exploit the task-specific domain knowledge via the oval shapes of OD & OC, where a differentiable vCDR estimating layer is proposed. Furthermore, without additional annotations, the supervision on vCDR serves as weakly-supervisions for segmentation tasks. Experiments on six large-scale datasets demonstrate our model's superior performance on OD & OC segmentation and vCDR estimation. The implementation code has been made available.https://github.com/smallmax00/Dual_Adaptive_Graph_Reasoning.
Yanda Meng, Hongrun Zhang, Yitian Zhao, Dongxu Gao, Barbra Hamill, Godhuli Patri, Tunde Peto, Savita Madhusudhan, Yalin Zheng
IEEE Trans. Medical Imaging3
2023 3D Human Pose and Shape Reconstruction From Videos via Confidence-Aware Temporal Feature Aggregation
abstract
Estimating 3D human body shapes and poses from videos is a challenging computer vision task. The intrinsic temporal information embedded in adjacent frames is helpful in making accurate estimations. Existing approaches learn temporal features of the target frames simply by aggregating features of their adjacent frames, using off-the-shelf deep neural networks. Consequently these approaches cannot explicitly and effectively use the correlations between adjacent frames to help infer the parameters of the target frames. In this paper, we propose a novel framework that can measure the correlations amongst adjacent frames in the form of an estimated confidence metric. The confidence value will indicate to what extent the adjacent frames can help predict the target frames’ 3D shapes and poses. Based on the estimated confidence values, temporally aggregated features are then obtained by adaptively allocating different weights to the temporal predicted features from the adjacent frames. The final 3D shapes and poses are estimated by regressing from the temporally aggregated features. Experimental results on three benchmark datasets show that the proposed method outperforms state-of-the-art approaches (even without the motion priors involved in training). In particular, the proposed method is more robust against corrupted frames.
Hongrun Zhang, Yanda Meng, Yitian Zhao, Xuesheng Qian, Yihong Qiao, Xiaoyun Yang, Yalin Zheng
IEEE Trans. Multim.3
2022 Topology-Aware Learning for Semi-supervised Cross-domain Retinal Artery/Vein Classification
Jianyang Xie, Yonghuai Liu, Huaying Hao, Lijun Guo, Jiong Zhang 0004, Yitian Zhao
CGI7
2022 DTFD-MIL: Double-Tier Feature Distillation Multiple Instance Learning for Histopathology Whole Slide Image Classification
abstract
Multiple instance learning (MIL) has been increasingly used in the classification of histopathology whole slide images (WSIs). However, MIL approaches for this specific classification problem still face unique challenges, particularly those related to small sample cohorts. In these, there are limited number of WSI slides (bags), while the resolution of a single WSI is huge, which leads to a large number of patches (instances) cropped from this slide. To address this issue, we propose to virtually enlarge the number of bags by introducing the concept of pseudo-bags, on which a double-tier MIL framework is built to effectively use the intrinsic features. Besides, we also contribute to deriving the instance probability under the framework of attentionbased MIL, and utilize the derivation to help construct and analyze the proposed framework. The proposed method outperforms other latest methods on the CAMELYON-16 by substantially large margins, and is also better in performance on the TCGA lung cancer dataset. The proposed framework is ready to be extended for wider MIL applications. The code is available at: https://github. com/hrzhang1123/DTFD-MIL.
Hongrun Zhang, Yanda Meng, Yitian Zhao, Yihong Qiao, Xiaoyun Yang, Sarah E. Coupland, Yalin Zheng
CVPR3
2022 Spatial-Context-Aware Deep Neural Network for Multi-Class Image Classification
abstract
Multi-label image classification is a fundamental but challenging task in computer vision. Over the past few decades, solutions exploring relationships between semantic labels have made great progress. However, the underlying spatial-contextual information of labels is under-exploited. To tackle this problem, a spatial-context-aware deep neural network is proposed to predict labels taking into account both semantic and spatial information. This proposed framework is evaluated on Microsoft COCO and PASCAL VOC, two widely used benchmark datasets for image multi-labelling. The results show that the proposed approach is superior to the state-of-the-art solutions on dealing with the multi-label image classification problem.
Jialu Zhang 0003, Qian Zhang 0018, Jianfeng Ren, Yitian Zhao, Jiang Liu 0001
ICASSP4
2022 Unsupervised Lesion-Aware Transfer Learning for Diabetic Retinopathy Grading in Ultra-Wide-Field Fundus Photography
Yanmiao Bai, Jinkui Hao, Huazhu Fu, Xinting Ge, Jiang Liu 0001, Yitian Zhao, Jiong Zhang 0004
MICCAI (2)7
2022 NerveFormer: A Cross-Sample Aggregation Network for Corneal Nerve Segmentation
Lei Mou, Shaodong Ma, Huazhu Fu, Lijun Guo, Yalin Zheng, Jiong Zhang 0004, Yitian Zhao
MICCAI (4)8
2022 Structure-Consistent Restoration Network for Cataract Fundus Image Enhancement
Heng Li 0010, Haofeng Liu, Huazhu Fu, Hai Shu, Yitian Zhao, Jiang Liu 0001
MICCAI (2)5
2022 Shape-Aware Weakly/Semi-Supervised Optic Disc and Cup Segmentation with Regional/Marginal Consistency
Yanda Meng, Xu Chen 0030, Hongrun Zhang, Yitian Zhao, Dongxu Gao, Barbra Hamill, Godhuli Patri, Tunde Peto, Savita Madhusudhan, Yalin Zheng
MICCAI (4)4
2022 Screening of Dementia on OCTA Images via Multi-projection Consistency and Complementarity
Heng Li 0010, Zunjie Xiao, Huazhu Fu, Yitian Zhao, Richu Jin, William Robert Kwapong, Hanpei Miao, Jiang Liu 0001
MICCAI (2)5
2022 Unsupervised Multi-View CNN for Salient View Selection and 3D Interest Point Detection
Ran Song 0001, Wei Zhang 0021, Yitian Zhao, Yonghuai Liu
Int. J. Comput. Vis.3
2022 Uncertainty-guided graph attention network for parapneumonic effusion diagnosis
Jinkui Hao, Jiang Liu 0001, Ella Grishikashvili Pereira, Ri Liu, Jiong Zhang 0004, Yangfan Zhang, Jianjun Zheng, Jingfeng Zhang, Yonghuai Liu, Yitian Zhao
Medical Image Anal.12
2022 3D vessel-like structure segmentation in medical images by an edge-reinforced network
Likun Xia, Hao Zhang 0113, Yufei Wu 0013, Ran Song 0001, Yuhui Ma, Lei Mou, Jiang Liu 0001, Ming Ma 0004, Yitian Zhao
Medical Image Anal.10
2022 Sparse-Based Domain Adaptation Network for OCTA Image Super-Resolution Reconstruction
abstract
Retinal Optical Coherence Tomography Angiography (OCTA) with high-resolution is important for the quantification and analysis of retinal vasculature. However, the resolution of OCTA images is inversely proportional to the field of view at the same sampling frequency, which is not conducive to clinicians for analyzing larger vascular areas. In this paper, we propose a novel Sparse-based domain Adaptation Super-Resolution network (SASR) for the reconstruction of realistic [Formula: see text]/low-resolution (LR) OCTA images to high-resolution (HR) representations. To be more specific, we first perform a simple degradation of the [Formula: see text]/high-resolution (HR) image to obtain the synthetic LR image. An efficient registration method is then employed to register the synthetic LR with its corresponding [Formula: see text] image region within the [Formula: see text] image to obtain the cropped realistic LR image. We then propose a multi-level super-resolution model for the fully-supervised reconstruction of the synthetic data, guiding the reconstruction of the realistic LR images through a generative-adversarial strategy that allows the synthetic and realistic LR images to be unified in the feature domain. Finally, a novel sparse edge-aware loss is designed to dynamically optimize the vessel edge structure. Extensive experiments on two OCTA sets have shown that our method performs better than state-of-the-art super-resolution reconstruction methods. In addition, we have investigated the performance of the reconstruction results on retina structure segmentations, which further validate the effectiveness of our approach.
Huaying Hao, Dan Zhang 0026, Qifeng Yan, Jiong Zhang 0004, Yue Liu 0005, Yitian Zhao
IEEE J. Biomed. Health Informatics7
2022 Multi-Scale Interactive Network With Artery/Vein Discriminator for Retinal Vessel Classification
abstract
Automatic classification of retinal arteries and veins plays an important role in assisting clinicians to diagnosis cardiovascular and eye-related diseases. However, due to the high degree of anatomical variation across the population, and the presence of inconsistent labels by the subjective judgment of annotators in available training data, most of existing methods generally suffer from blood vessel discontinuity and arteriovenous confusion, the artery/vein (A/V) classification task still faces great challenges. In this work, we propose a multi-scale interactive network with A/V discriminator for retinal artery and vein recognition, which can reduce the arteriovenous confusion and alleviate the disturbance of noisy label. A multi-scale interaction (MI) module is designed in encoder for realizing the cross-space multi-scale features interaction of fundus images, effectively integrate high-level and low-level context information. In particular, we also design an ingenious A/V discriminator (AVD) that utilizes the independent and shared information between arteries and veins, and combine with topology loss, to further strengthen the learning ability of model to resolve the arteriovenous confusion. In addition, we adopt a sample re-weighting (SW) strategy to effectively alleviate the disturbance from data labeling errors. The proposed model is verified on three publicly available fundus image datasets (AV-DRIVE, HRF, LES-AV) and a private dataset. We achieve the accuracy of 97.47%, 96.91%, 97.79%, and 98.18% respectively on these four datasets. Extensive experimental results demonstrate that our method achieves competitive performance compared with state-of-the-art methods for A/V classification. To address the problem of training data scarcity, we publicly release 100 fundus images with A/V annotations to promote relevant research in the community.
Jingfei Hu, Hua Wang 0014, Zhaohui Cao, Lei Mou, Yitian Zhao, Jicong Zhang
IEEE J. Biomed. Health Informatics6
2022 Explainable Diabetic Retinopathy Detection and Retinal Image Generation
abstract
Though deep learning has shown successful performance in classifying the label and severity stage of certain diseases, most of them give few explanations on how to make predictions. Inspired by Koch's Postulates, the foundation in evidence-based medicine (EBM) to identify the pathogen, we propose to exploit the interpretability of deep learning application in medical diagnosis. By isolating neuron activation patterns from a diabetic retinopathy (DR) detector and visualizing them, we can determine the symptoms that the DR detector identifies as evidence to make prediction. To be specific, we first define novel pathological descriptors using activated neurons of the DR detector to encode both spatial and appearance information of lesions. Then, to visualize the symptom encoded in the descriptor, we propose Patho-GAN, a new network to synthesize medically plausible retinal images. By manipulating these descriptors, we could even arbitrarily control the position, quantity, and categories of generated lesions. We also show that our synthesized images carry the symptoms directly related to diabetic retinopathy diagnosis. Our generated images are both qualitatively and quantitatively superior to the ones by previous methods. Besides, compared to existing methods that take hours to generate an image, our second level speed endows the potential to be an effective solution for data augmentation.
Yuhao Niu, Lin Gu 0003, Yitian Zhao, Feng Lu 0005
IEEE J. Biomed. Health Informatics3
2022 Hybrid Variation-Aware Network for Angle-Closure Assessment in AS-OCT
abstract
Automatic angle-closure assessment in Anterior Segment OCT (AS-OCT) images is an important task for the screening and diagnosis of glaucoma, and the most recent computer-aided models focus on a binary classification of anterior chamber angles (ACA) in AS-OCT, i.e., open-angle and angle-closure. In order to assist clinicians who seek better to understand the development of the spectrum of glaucoma types, a more discriminating three-class classification scheme was suggested, i.e., the classification of ACA was expended to include open-, appositional- and synechial angles. However, appositional and synechial angles display similar appearances in an AS-OCT image, which makes classification models struggle to differentiate angle-closure subtypes based on static AS-OCT images. In order to tackle this issue, we propose a 2D-3D Hybrid Variation-aware Network (HV-Net) for open-appositional-synechial ACA classification from AS-OCT imagery. Specifically, taking into account clinical priors, we first reconstruct the 3D iris surface from an AS-OCT sequence, and obtain the geometrical characteristics necessary to provide global shape information. 2D AS-OCT slices and 3D iris representations are then fed into our HV-Net to extract cross-sectional appearance features and iris morphological features, respectively. To achieve similar results to those of dynamic gonioscopy examination, which is the current gold standard for diagnostic angle assessment, the paired AS-OCT images acquired in dark and light illumination conditions are used to obtain an accurate characterization of configurational changes in ACAs and iris shapes, using a Variation-aware Block. In addition, an annealing loss function was introduced to optimize our model, so as to encourage the sub-networks to map the inputs into the more conducive spaces to extract dark-to-light variation representations, while retaining the discriminative power of the learned features. The proposed model is evaluated across 1584 paired AS-OCT samples, and it has demonstrated its superiority in classifying open-, appositional- and synechial angles.
Jinkui Hao, Fei Li 0021, Huaying Hao, Huazhu Fu, Yanwu Xu 0001, Risa Higashita, Xiulan Zhang, Jiang Liu 0001, Yitian Zhao
IEEE Trans. Medical Imaging9
2022 Retinal Structure Detection in OCTA Image via Voting-Based Multitask Learning
abstract
Automated detection of retinal structures, such as retinal vessels (RV), the foveal avascular zone (FAZ), and retinal vascular junctions (RVJ), are of great importance for understanding diseases of the eye and clinical decision-making. In this paper, we propose a novel Voting-based Adaptive Feature Fusion multi-task network (VAFF-Net) for joint segmentation, detection, and classification of RV, FAZ, and RVJ in optical coherence tomography angiography (OCTA). A task-specific voting gate module is proposed to adaptively extract and fuse different features for specific tasks at two levels: features at different spatial positions from a single encoder, and features from multiple encoders. In particular, since the complexity of the microvasculature in OCTA images makes simultaneous precise localization and classification of retinal vascular junctions into bifurcation/crossing a challenging task, we specifically design a task head by combining the heatmap regression and grid classification. We take advantage of three different en face angiograms from various retinal layers, rather than following existing methods that use only a single en face. We carry out extensive experiments on three OCTA datasets acquired using different imaging devices, and the results demonstrate that the proposed method performs on the whole better than either the state-of-the-art single-purpose methods or existing multi-task learning solutions. We also demonstrate that our multi-task learning method generalizes across other imaging modalities, such as color fundus photography, and may potentially be used as a general multi-task learning tool. We also construct three datasets for multiple structure detection, and part of these datasets with the source code and evaluation benchmark have been released for public access.
Jinkui Hao, Ting Shen, Xueli Zhu 0002, Yonghuai Liu, Ardhendu Behera, Dan Zhang 0026, Bang Chen, Jiang Liu 0001, Jiong Zhang 0004, Yitian Zhao
IEEE Trans. Medical Imaging10
2022 An Annotation-Free Restoration Network for Cataractous Fundus Images
abstract
Cataracts are the leading cause of vision loss worldwide. Restoration algorithms are developed to improve the readability of cataract fundus images in order to increase the certainty in diagnosis and treatment for cataract patients. Unfortunately, the requirement of annotation limits the application of these algorithms in clinics. This paper proposes a network to annotation-freely restore cataractous fundus images (ArcNet) so as to boost the clinical practicability of restoration. Annotations are unnecessary in ArcNet, where the high-frequency component is extracted from fundus images to replace segmentation in the preservation of retinal structures. The restoration model is learned from the synthesized images and adapted to real cataract images. Extensive experiments are implemented to verify the performance and effectiveness of ArcNet. Favorable performance is achieved using ArcNet against state-of-the-art algorithms, and the diagnosis of ocular fundus diseases in cataract patients is promoted by ArcNet. The capability of properly restoring cataractous images in the absence of annotated data promises the proposed algorithm outstanding clinical practicability.
Heng Li 0010, Haofeng Liu, Huazhu Fu, Yitian Zhao, Hanpei Miao, Jiang Liu 0001
IEEE Trans. Medical Imaging5
2022 Graph-Based Region and Boundary Aggregation for Biomedical Image Segmentation
abstract
Segmentation is a fundamental task in biomedical image analysis. Unlike the existing region-based dense pixel classification methods or boundary-based polygon regression methods, we build a novel graph neural network (GNN) based deep learning framework with multiple graph reasoning modules to explicitly leverage both region and boundary features in an end-to-end manner. The mechanism extracts discriminative region and boundary features, referred to as initialized region and boundary node embeddings, using a proposed Attention Enhancement Module (AEM). The weighted links between cross-domain nodes (region and boundary feature domains) in each graph are defined in a data-dependent way, which retains both global and local cross-node relationships. The iterative message aggregation and node update mechanism can enhance the interaction between each graph reasoning module's global semantic information and local spatial characteristics. Our model, in particular, is capable of concurrently addressing region and boundary feature reasoning and aggregation at several different feature levels due to the proposed multi-level feature node embeddings in different parallel graph reasoning modules. Experiments on two types of challenging datasets demonstrate that our method outperforms state-of-the-art approaches for segmentation of polyps in colonoscopy images and of the optic disc and optic cup in colour fundus images. The trained models will be made available at: https://github.com/smallmax00/Graph_Region_Boudnary.
Yanda Meng, Hongrun Zhang, Yitian Zhao, Xiaoyun Yang, Yihong Qiao, Ian J. C. MacCormick, Xiaowei Huang 0001, Yalin Zheng
IEEE Trans. Medical Imaging3
2022 DeepGrading: Deep Learning Grading of Corneal Nerve Tortuosity
abstract
Accurate estimation and quantification of the corneal nerve fiber tortuosity in corneal confocal microscopy (CCM) is of great importance for disease understanding and clinical decision-making. However, the grading of corneal nerve tortuosity remains a great challenge due to the lack of agreements on the definition and quantification of tortuosity. In this paper, we propose a fully automated deep learning method that performs image-level tortuosity grading of corneal nerves, which is based on CCM images and segmented corneal nerves to further improve the grading accuracy with interpretability principles. The proposed method consists of two stages: 1) A pre-trained feature extraction backbone over ImageNet is fine-tuned with a proposed novel bilinear attention (BA) module for the prediction of the regions of interest (ROIs) and coarse grading of the image. The BA module enhances the ability of the network to model long-range dependencies and global contexts of nerve fibers by capturing second-order statistics of high-level features. 2) An auxiliary tortuosity grading network (AuxNet) is proposed to obtain an auxiliary grading over the identified ROIs, enabling the coarse and additional gradings to be finally fused together for more accurate final results. The experimental results show that our method surpasses existing methods in tortuosity grading, and achieves an overall accuracy of 85.64% in four-level classification. We also validate it over a clinical dataset, and the statistical analysis demonstrates a significant difference of tortuosity levels between healthy control and diabetes group. We have released a dataset with 1500 CCM images and their manual annotations of four tortuosity levels for public access. The code is available at: https://github.com/iMED-Lab/TortuosityGrading.
Lei Mou, Yonghuai Liu, Yalin Zheng, Peter Matthew, Pan Su 0001, Jiang Liu 0001, Jiong Zhang 0004, Yitian Zhao
IEEE Trans. Medical Imaging9
2021 BI-GCN: Boundary-Aware Input-Dependent Graph Convolution Network for Biomedical Image Segmentation
Yanda Meng, Hongrun Zhang, Dongxu Gao, Yitian Zhao, Xiaoyun Yang, Xuesheng Qian, Xiaowei Huang 0001, Yalin Zheng
BMVC4
2021 Mesh Saliency: An Independent Perceptual Measure or a Derivative of Image Saliency?
abstract
While mesh saliency aims to predict regional importance of 3D surfaces in agreement with human visual perception and is well researched in computer vision and graphics, latest work with eye-tracking experiments shows that state-of-the-art mesh saliency methods remain poor at predicting human fixations. Cues emerging prominently from these experiments suggest that mesh saliency might associate with the saliency of 2D natural images. This paper proposes a novel deep neural network for learning mesh saliency using image saliency ground truth to 1) investigate whether mesh saliency is an independent perceptual measure or just a derivative of image saliency and 2) provide a weakly supervised method for more accurately predicting mesh saliency. Through extensive experiments, we not only demonstrate that our method outperforms the current state-of-the-art mesh saliency method by 116% and 21% in terms of linear correlation coefficient and AUC respectively, but also reveal that mesh saliency is intrinsically related with both image saliency and object categorical information. Codes are available at https://github.com/rsong/MIMO-GAN.
Ran Song 0001, Wei Zhang 0021, Yitian Zhao, Yonghuai Liu, Paul L. Rosin
CVPR3
2021 Spatial Uncertainty-Aware Semi-Supervised Crowd Counting
abstract
Semi-supervised approaches for crowd counting attract attention, as the fully supervised paradigm is expensive and laborious due to its request for a large number of images of dense crowd scenarios and their annotations. This paper proposes a spatial uncertainty-aware semi-supervised approach via regularized surrogate task (binary segmentation) for crowd counting problems. Different from existing semi-supervised learning-based crowd counting methods, to exploit the unlabeled data, our proposed spatial uncertainty-aware teacher-student framework focuses on high confident regions’ information while addressing the noisy supervision from the unlabeled data in an end-to-end manner. Specifically, we estimate the spatial uncertainty maps from the teacher model’s surrogate task to guide the feature learning of the main task (density regression) and the surrogate task of the student model at the same time. Besides, we introduce a simple yet effective differential transformation layer to enforce the inherent spatial consistency regularization between the main task and the surrogate task in the student model, which helps the surrogate task to yield more reliable predictions and generates high-quality uncertainty maps. Thus, our model can also address the task-level perturbation problems that occur spatial inconsistency between the primary and surrogate tasks in the student model. Experimental results on four challenging crowd counting datasets demonstrate that our method achieves superior performance to the state-of-the-art semi-supervised methods. Code is available at : https://github.com/smallmax00/SUA_crowd_counting
Yanda Meng, Hongrun Zhang, Yitian Zhao, Xiaoyun Yang, Xuesheng Qian, Xiaowei Huang 0001, Yalin Zheng
ICCV3
2021 Learning Unsupervised Parameter-Specific Affine Transformation for Medical Images Registration
Xu Chen 0030, Yanda Meng, Yitian Zhao, Rachel Williams, Srinivasa R. Vallabhaneni, Yalin Zheng
MICCAI (4)3
2021 Cross-Domain Depth Estimation Network for 3D Vessel Reconstruction in OCT Angiography
Yonghuai Liu, Jiong Zhang 0004, Jianyang Xie, Yalin Zheng, Jiang Liu 0001, Yitian Zhao
MICCAI (8)7
2021 A Decision Tree-Initialised Neuro-fuzzy Approach for Clinical Decision Support
abstract
Apart from the need for superior accuracy, healthcare applications of intelligent systems also demand the deployment of interpretable machine learning models which allow clinicians to interrogate and validate extracted medical knowledge. Fuzzy rule-based models are generally considered interpretable that are able to reflect the associations between medical conditions and associated symptoms, through the use of linguistic if-then statements. Systems built on top of fuzzy sets are of particular appealing to medical applications since they enable the tolerance of vague and imprecise concepts that are often embedded in medical entities such as symptom description and test results. They facilitate an approximate reasoning framework which mimics human reasoning and supports the linguistic delivery of medical expertise often expressed in statements such as 'weight low' or 'glucose level high' while describing symptoms. This paper proposes an approach by performing data-driven learning of accurate and interpretable fuzzy rule bases for clinical decision support. The approach starts with the generation of a crisp rule base through a decision tree learning mechanism, capable of capturing simple rule structures. The crisp rule base is then transformed into a fuzzy rule base, which forms the input to the framework of adaptive network-based fuzzy inference system (ANFIS), thereby further optimising the parameters of both rule antecedents and consequents. Experimental studies on popular medical data benchmarks demonstrate that the proposed work is able to learn compact rule bases involving simple rule antecedents, with statistically better or comparable performance to those achieved by state-of-the-art fuzzy classifiers.
Tianhua Chen, Changjing Shang, Pan Su 0001, Elpida Keravnou-Papailiou, Yitian Zhao, Grigoris Antoniou, Qiang Shen 0001
Artif. Intell. Medicine5
2021 Angle-closure assessment in anterior segment OCT images via deep learning
Huaying Hao, Yitian Zhao, Qifeng Yan, Risa Higashita, Jiong Zhang 0004, Yifan Zhao 0001, Yanwu Xu 0001, Fei Li 0021, Xiulan Zhang, Jiang Liu 0001
Medical Image Anal.2
2021 CS2-Net: Deep learning segmentation of curvilinear structures in medical imaging
Lei Mou, Yitian Zhao, Huazhu Fu, Yonghuai Liu, Jun Cheng 0003, Yalin Zheng, Pan Su 0001, Jianlong Yang, Li Chen 0011, Alejandro F. Frangi, Masahiro Akiba, Jiang Liu 0001
Medical Image Anal.2
2021 Interwoven texture-based description of interest points in images
Morteza Ghahremani, Yitian Zhao, Bernard Tiddeman, Yonghuai Liu
Pattern Recognit.2
2021 Understanding adversarial attacks on deep learning based medical image analysis systems
Xingjun Ma, Yuhao Niu, Lin Gu 0003, Yisen Wang 0001, Yitian Zhao, James Bailey 0001, Feng Lu 0005
Pattern Recognit.5
2021 A Miniaturized Active Thermography System to Inspect Composite Laminates
abstract
With the rapid increase of the integration and complexity of industrial components, the inaccessibility and inapplicability of existing nondestructive testing devices have become a bottleneck for in situ inspection of these objects. This article introduces a miniaturized active thermography system featured with a small-size, low-resolution, and low-cost thermal sensor, where two optional excitation sources including flash and laser are integrated. Dedicated data analysis approaches to evaluate defects are proposed considering the degraded signal quality. Three carbon fiber-reinforced polymer laminates with a variety of defects are evaluated quantitatively and qualitatively using the proposed system by comparing with two existing nonminiaturized inspection systems. The results show that the proposed system can work effectively for the degradation assessment of composite laminates. Even with the technical limitations that affect the detectability, for instance, the low pixel resolution, this technique will play an important role to inspect components featured with geometrically intricate space.
Weixiang Du, Yitian Zhao, Adisorn Sirikham, Sri Addepalli, Yifan Zhao 0001
IEEE Trans. Ind. Informatics3
2021 ROSE: A Retinal OCT-Angiography Vessel Segmentation Dataset and New Model
abstract
Optical Coherence Tomography Angiography (OCTA) is a non-invasive imaging technique that has been increasingly used to image the retinal vasculature at capillary level resolution. However, automated segmentation of retinal vessels in OCTA has been under-studied due to various challenges such as low capillary visibility and high vessel complexity, despite its significance in understanding many vision-related diseases. In addition, there is no publicly available OCTA dataset with manually graded vessels for training and validation of segmentation algorithms. To address these issues, for the first time in the field of retinal image analysis we construct a dedicated Retinal OCTA SEgmentation dataset (ROSE), which consists of 229 OCTA images with vessel annotations at either centerline-level or pixel level. This dataset with the source code has been released for public access to assist researchers in the community in undertaking research in related topics. Secondly, we introduce a novel split-based coarse-to-fine vessel segmentation network for OCTA images (OCTA-Net), with the ability to detect thick and thin vessels separately. In the OCTA-Net, a split-based coarse segmentation module is first utilized to produce a preliminary confidence map of vessels, and a split-based refined segmentation module is then used to optimize the shape/contour of the retinal microvasculature. We perform a thorough evaluation of the state-of-the-art vessel segmentation models and our OCTA-Net on the constructed ROSE dataset. The experimental results demonstrate that our OCTA-Net yields better vessel segmentation performance in OCTA than both traditional and other deep learning methods. In addition, we provide a fractal dimension analysis on the segmented microvasculature, and the statistical analysis demonstrates significant differences between the healthy control and Alzheimer's Disease group. This consolidates that the analysis of retinal microvasculature may offer a new scheme to study various neurodegenerative diseases.
Yuhui Ma, Huaying Hao, Jianyang Xie, Huazhu Fu, Jiong Zhang 0004, Jianlong Yang, Jiang Liu 0001, Yalin Zheng, Yitian Zhao
IEEE Trans. Medical Imaging10
2021 Structure and Illumination Constrained GAN for Medical Image Enhancement
abstract
The development of medical imaging techniques has greatly supported clinical decision making. However, poor imaging quality, such as non-uniform illumination or imbalanced intensity, brings challenges for automated screening, analysis and diagnosis of diseases. Previously, bi-directional GANs (e.g., CycleGAN), have been proposed to improve the quality of input images without the requirement of paired images. However, these methods focus on global appearance, without imposing constraints on structure or illumination, which are essential features for medical image interpretation. In this paper, we propose a novel and versatile bi-directional GAN, named Structure and illumination constrained GAN (StillGAN), for medical image quality enhancement. Our StillGAN treats low- and high-quality images as two distinct domains, and introduces local structure and illumination constraints for learning both overall characteristics and local details. Extensive experiments on three medical image datasets (e.g., corneal confocal microscopy, retinal color fundus and endoscopy images) demonstrate that our method performs better than both conventional methods and other deep learning-based methods. In addition, we have investigated the impact of the proposed method on different medical image analysis and clinical tasks such as nerve segmentation, tortuosity grading, fovea localization and disease classification.
Yuhui Ma, Jiang Liu 0001, Yonghuai Liu, Huazhu Fu, Jun Cheng 0003, Yufei Wu 0013, Jiong Zhang 0004, Yitian Zhao
IEEE Trans. Medical Imaging10
2020 Regression of Instance Boundary by Aggregated CNN and GCN
Yanda Meng, Dongxu Gao, Yitian Zhao, Xiaoyun Yang, Xiaowei Huang 0001, Yalin Zheng
ECCV (8)4
2020 Unsupervised Multi-view CNN for Salient View Selection of 3D Objects and Scenes
Ran Song 0001, Wei Zhang 0021, Yitian Zhao, Yonghuai Liu
ECCV (19)3
2020 Reconstruction and Quantification of 3D Iris Surface for Angle-Closure Glaucoma Detection in Anterior Segment OCT
Jinkui Hao, Huazhu Fu, Yanwu Xu 0001, Fei Li 0021, Xiulan Zhang, Jiang Liu 0001, Yitian Zhao
MICCAI (5)8
2020 Open-Appositional-Synechial Anterior Chamber Angle Classification in AS-OCT Sequences
Huaying Hao, Huazhu Fu, Yanwu Xu 0001, Jianlong Yang, Fei Li 0021, Xiulan Zhang, Jiang Liu 0001, Yitian Zhao
MICCAI (5)8
2020 Cycle Structure and Illumination Constrained GAN for Medical Image Enhancement
Yuhui Ma, Yonghuai Liu, Jun Cheng 0003, Yalin Zheng, Morteza Ghahremani, Honghan Chen, Jiang Liu 0001, Yitian Zhao
MICCAI (2)8
2020 CNN-GCN Aggregation Enabled Boundary Regression for Biomedical Image Segmentation
Yanda Meng, Dongxu Gao, Yitian Zhao, Xiaoyun Yang, Xiaowei Huang 0001, Yalin Zheng
MICCAI (4)4
2020 Classification of Retinal Vessels into Artery-Vein in OCT Angiography Guided by Fundus Images
Jianyang Xie, Yonghuai Liu, Yalin Zheng, Pan Su 0001, Jian Yang 0009, Jiang Liu 0001, Yitian Zhao
MICCAI (6)8
2020 Cerebrovascular Segmentation in MRA via Reverse Edge Attention Network
Hao Zhang 0113, Likun Xia, Ran Song 0001, Jianlong Yang, Huaying Hao, Jiang Liu 0001, Yitian Zhao
MICCAI (6)7
2020 Speckle reduction of OCT via super resolution reconstruction and its application on retinal layer segmentation
Qifeng Yan, Bang Chen, Jun Cheng 0003, Jianlong Yang, Jiang Liu 0001, Yitian Zhao
Artif. Intell. Medicine8
2020 Introducing the GEV Activation Function for Highly Unbalanced Data to Develop COVID-19 Diagnostic Models
abstract
Fast and accurate diagnosis is essential for the efficient and effective control of the COVID-19 pandemic that is currently disrupting the whole world. Despite the prevalence of the COVID-19 outbreak, relatively few diagnostic images are openly available to develop automatic diagnosis algorithms. Traditional deep learning methods often struggle when data is highly unbalanced with many cases in one class and only a few cases in another; new methods must be developed to overcome this challenge. We propose a novel activation function based on the generalized extreme value (GEV) distribution from extreme value theory, which improves performance over the traditional sigmoid activation function when one class significantly outweighs the other. We demonstrate the proposed activation function on a publicly available dataset and externally validate on a dataset consisting of 1,909 healthy chest X-rays and 84 COVID-19 X-rays. The proposed method achieves an improved area under the receiver operating characteristic (DeLong's p-value < 0.05) compared to the sigmoid activation. Our method is also demonstrated on a dataset of healthy and pneumonia vs. COVID-19 X-rays and a set of computerized tomography images, achieving improved sensitivity. The proposed GEV activation function significantly improves upon the previously used sigmoid activation for binary classification. This new paradigm is expected to play a significant role in the fight against COVID-19 and other diseases, with relatively few training cases available.
Joshua Bridge, Yanda Meng, Yitian Zhao, Mingfeng Zhao, Renrong Sun, Yalin Zheng
IEEE J. Biomed. Health Informatics3
2020 Automatic Segmentation and Visualization of Choroid in OCT with Knowledge Infused Deep Learning
abstract
The choroid provides oxygen and nourishment to the outer retina thus is related to the pathology of various ocular diseases. Optical coherence tomography (OCT) is advantageous in visualizing and quantifying the choroid in vivo. However, its application in the study of the choroid is still limited for two reasons. (1) The lower boundary of the choroid (choroid-sclera interface) in OCT is fuzzy, which makes the automatic segmentation difficult and inaccurate. (2) The visualization of the choroid is hindered by the vessel shadows from the superficial layers of the inner retina. In this paper, we propose to incorporate medical and imaging prior knowledge with deep learning to address these two problems. We propose a biomarker-infused global-to-local network (Bio-Net) for the choroid segmentation, which not only regularizes the segmentation via predicted choroid thickness, but also leverages a global-to-local segmentation strategy to provide global structure information and suppress overfitting. For eliminating the retinal vessel shadows, we propose a deep-learning pipeline, which firstly locate the shadows using their projection on the retinal pigment epithelium layer, then the contents of the choroidal vasculature at the shadow locations are predicted with an edge-to-texture generative adversarial inpainting network. The results show our method outperforms the existing methods on both tasks. We further apply the proposed method in a clinical prospective study for understanding the pathology of glaucoma, which demonstrates its capacity in detecting the structure and vascular changes of the choroid related to the elevation of intra-ocular pressure.
Huihong Zhang, Jianlong Yang, Kang Zhou 0001, Fei Li 0021, Yitian Zhao, Xiulan Zhang, Jiang Liu 0001
IEEE J. Biomed. Health Informatics6
2020 Dense Dilated Network With Probability Regularized Walk for Vessel Detection
abstract
The detection of retinal vessel is of great importance in the diagnosis and treatment of many ocular diseases. Many methods have been proposed for vessel detection. However, most of the algorithms neglect the connectivity of the vessels, which plays an important role in the diagnosis. In this paper, we propose a novel method for retinal vessel detection. The proposed method includes a dense dilated network to get an initial detection of the vessels and a probability regularized walk algorithm to address the fracture issue in the initial detection. The dense dilated network integrates newly proposed dense dilated feature extraction blocks into an encoder-decoder structure to extract and accumulate features at different scales. A multi-scale Dice loss function is adopted to train the network. To improve the connectivity of the segmented vessels, we also introduce a probability regularized walk algorithm to connect the broken vessels. The proposed method has been applied on three public data sets: DRIVE, STARE and CHASE_DB1. The results show that the proposed method outperforms the state-of-the-art methods in accuracy, sensitivity, specificity and also area under receiver operating characteristic curve.
Lei Mou, Li Chen 0011, Jun Cheng 0003, Zaiwang Gu, Yitian Zhao, Jiang Liu 0001
IEEE Trans. Medical Imaging5
2020 Retinal Vascular Network Topology Reconstruction and Artery/Vein Classification via Dominant Set Clustering
abstract
The estimation of vascular network topology in complex networks is important in understanding the relationship between vascular changes and a wide spectrum of diseases. Automatic classification of the retinal vascular trees into arteries and veins is of direct assistance to the ophthalmologist in terms of diagnosis and treatment of eye disease. However, it is challenging due to their projective ambiguity and subtle changes in appearance, contrast, and geometry in the imaging process. In this paper, we propose a novel method that is capable of making the artery/vein (A/V) distinction in retinal color fundus images based on vascular network topological properties. To this end, we adapt the concept of dominant set clustering and formalize the retinal blood vessel topology estimation and the A/V classification as a pairwise clustering problem. The graph is constructed through image segmentation, skeletonization, and identification of significant nodes. The edge weight is defined as the inverse Euclidean distance between its two end points in the feature space of intensity, orientation, curvature, diameter, and entropy. The reconstructed vascular network is classified into arteries and veins based on their intensity and morphology. The proposed approach has been applied to five public databases, namely INSPIRE, IOSTAR, VICAVR, DRIVE, and WIDE, and achieved high accuracies of 95.1%, 94.2%, 93.8%, 91.1%, and 91.0%, respectively. Furthermore, we have made manual annotations of the blood vessel topologies for INSPIRE, IOSTAR, VICAVR, and DRIVE datasets, and these annotations are released for public access so as to facilitate researchers in the community.
Yitian Zhao, Yonghuai Liu, Jianyang Xie, Huaizhong Zhang, Yalin Zheng, Yifan Zhao 0001, Yangchun Zhao, Pan Su 0001, Jiang Liu 0001
IEEE Trans. Medical Imaging1
2020 Imaging of Nonlinear and Dynamic Functional Brain Connectivity Based on EEG Recordings With the Application on the Diagnosis of Alzheimer's Disease
abstract
Since age is the most significant risk factor for the development of Alzheimer's disease (AD), it is important to understand the effect of normal ageing on brain network characteristics before we can accurately diagnose the condition based on information derived from resting state electroencephalogram (EEG) recordings, aiming to detect brain network disruption. This article proposes a novel brain functional connectivity imaging method, particularly targeting the contribution of nonlinear dynamics of functional connectivity, on distinguishing participants with AD from healthy controls (HC). We describe a parametric method established upon a Nonlinear Finite Impulse Response model, and a revised orthogonal least squares algorithm used to estimate the linear, nonlinear and combined connectivity between any two EEG channels without fitting a full model. This approach, where linear and non-linear interactions and their spatial distribution and dynamics can be estimated independently, offered us the means to dissect the dynamic brain network disruption in AD from a new perspective and to gain some insight into the dynamic behaviour of brain networks in two age groups (above and below 70) with normal cognitive function. Although linear and stationary connectivity dominates the classification contributions, quantitative results have demonstrated that nonlinear and dynamic connectivity can significantly improve the classification accuracy, barring the group of participants below the age of 70, for resting state EEG recorded during eyes open. The developed approach is generic and can be used as a powerful tool to examine brain network characteristics and disruption in a user friendly and systematic way.
Yifan Zhao 0001, Yitian Zhao, Pholpat Durongbhan, Jiang Liu 0001, Stephen A. Billings, Panagiotis Zis, Zoe C. Unwin, Matteo De Marco, Annalena Venneri, Daniel Blackburn, Ptolemaios G. Sarrigiannis
IEEE Trans. Medical Imaging2
2020 Automated Tortuosity Analysis of Nerve Fibers in Corneal Confocal Microscopy
abstract
Precise characterization and analysis of corneal nerve fiber tortuosity are of great importance in facilitating examination and diagnosis of many eye-related diseases. In this paper we propose a fully automated method for image-level tortuosity estimation, comprising image enhancement, exponential curvature estimation, and tortuosity level classification. The image enhancement component is based on an extended Retinex model, which not only corrects imbalanced illumination and improves image contrast in an image, but also models noise explicitly to aid removal of imaging noise. Afterwards, we take advantage of exponential curvature estimation in the 3D space of positions and orientations to directly measure curvature based on the enhanced images, rather than relying on the explicit segmentation and skeletonization steps in a conventional pipeline usually with accumulated pre-processing errors. The proposed method has been applied over two corneal nerve microscopy datasets for the estimation of a tortuosity level for each image. The experimental results show that it performs better than several selected state-of-the-art methods. Furthermore, we have performed manual gradings at tortuosity level of four hundred and three corneal nerve microscopic images, and this dataset has been released for public access to facilitate other researchers in the community in carrying out further research on the same and related topics.
Yitian Zhao, Jiong Zhang 0004, Ella Grishikashvili Pereira, Yalin Zheng, Pan Su 0001, Jianyang Xie, Yifan Zhao 0001, Yonggang Shi, Jiang Liu 0001, Yonghuai Liu
IEEE Trans. Medical Imaging1
2020 Corrections to "Automated Tortuosity Analysis of Nerve Fibers in Corneal Confocal Microscopy"
abstract
In the above article[1], there were two errors in the printed article that the authors want to correct.
Yitian Zhao, Jiong Zhang 0004, Ella Grishikashvili Pereira, Yalin Zheng, Pan Su 0001, Jianyang Xie, Yifan Zhao 0001, Yonggang Shi, Jiang Liu 0001, Yonghuai Liu
IEEE Trans. Medical Imaging1
2019 Topology Reconstruction of Tree-Like Structure in Images via Structural Similarity Measure and Dominant Set Clustering
abstract
The reconstruction and analysis of tree-like topological structures in the biomedical images is crucial for biologists and surgeons to understand biomedical conditions and plan surgical procedures. The underlying tree-structure topology reveals how different curvilinear components are anatomically connected to each other. Existing automated topology reconstruction methods have great difficulty in identifying the connectivity when two or more curvilinear components cross or bifurcate, due to their projection ambiguity, imaging noise and low contrast. In this paper, we propose a novel curvilinear structural similarity measure to guide a dominant-set clustering approach to address this indispensable issue. The novel similarity measure takes into account both intensity and geometric properties in representing the curvilinear structure locally and globally, and group curvilinear objects at crossover points into different connected branches by dominant-set clustering. The proposed method is applicable to different imaging modalities, and quantitative and qualitative results on retinal vessel, plant root, and neuronal network datasets show that our methodology is capable of advancing the current state-of-the-art techniques.
Jianyang Xie, Yitian Zhao, Yonghuai Liu, Pan Su 0001, Yifan Zhao 0001, Jun Cheng 0003, Yalin Zheng, Jiang Liu 0001
CVPR2
2019 On the Application of Preaggregation Functions to Fuzzy Pattern Tree
abstract
Building transparent knowledge-based systems in the form of accurate and interpretable fuzzy rules is one of the significant applications of fuzzy set theory. The fuzzy connectives, i.e., T -norm/conorm, play the role of connecting fuzzy sets, which are essentially linguistic terms extracted from the knowledge embedded in a given data set. Fuzzy pattern tree is a recently proposed novel machine learning technique, which grows a hierarchical binary tree for each known class utilising conventional T -norms/conorms and aggregation operators. Preaggregation functions are recently proposed in the literature as a type of generalised aggregation functions, which have achieved successes in a number of applications. This paper proposes a preaggregation-based approach with application to the construction of fuzzy pattern tree. An experimental study is done to explore the performance of the fuzzy pattern tree where preaggregation functions are employed in comparison to that where conventional aggregation operators are utilised. Experimental results demonstrate that the performance of fuzzy pattern tree incorporated with the preaggregation function generated by Nilpotent minimum T -norm outperforms those with alternative preaggregation functions and the commonly used ordered weighted averaging operators.
Pan Su 0001, Tianhua Chen, Haoyu Mao, Jianyang Xie, Yitian Zhao, Jiang Liu 0001
FUZZ-IEEE5
2019 CS-Net: Channel and Spatial Attention Network for Curvilinear Structure Segmentation
Lei Mou, Yitian Zhao, Li Chen 0011, Jun Cheng 0003, Zaiwang Gu, Huaying Hao, Yalin Zheng, Alejandro F. Frangi, Jiang Liu 0001
MICCAI (1)2
2019 Exploiting Reliability-Guided Aggregation for the Assessment of Curvilinear Structure Tortuosity
Pan Su 0001, Yitian Zhao, Tianhua Chen, Jianyang Xie, Yifan Zhao 0001, Yalin Zheng, Jiang Liu 0001
MICCAI (4)2
2019 SkrGAN: Sketching-Rendering Unconditional Generative Adversarial Networks for Medical Image Synthesis
Tianyang Miller, Huazhu Fu, Yitian Zhao, Jun Cheng 0003, Mengjie Guo, Zaiwang Gu, Shenghua Gao, Jiang Liu 0001
MICCAI (4)3
2019 A mobilized automatic human body measure system using neural network
Likun Xia, Jian Yang 0009, Huiming Xu, Yitian Zhao, Yongtian Wang
Multim. Tools Appl.6
2019 CE-Net: Context Encoder Network for 2D Medical Image Segmentation
abstract
Medical image segmentation is an important step in medical image analysis. With the rapid development of a convolutional neural network in image processing, deep learning has been used for medical image segmentation, such as optic disc segmentation, blood vessel detection, lung segmentation, cell segmentation, and so on. Previously, U-net based approaches have been proposed. However, the consecutive pooling and strided convolutional operations led to the loss of some spatial information. In this paper, we propose a context encoder network (CE-Net) to capture more high-level information and preserve spatial information for 2D medical image segmentation. CE-Net mainly contains three major components: a feature encoder module, a context extractor, and a feature decoder module. We use the pretrained ResNet block as the fixed feature extractor. The context extractor module is formed by a newly proposed dense atrous convolution block and a residual multi-kernel pooling block. We applied the proposed CE-Net to different 2D medical image segmentation tasks. Comprehensive results show that the proposed method outperforms the original U-Net method and other state-of-the-art methods for optic disc segmentation, vessel detection, lung segmentation, cell contour segmentation, and retinal optical coherence tomography layer segmentation.
Zaiwang Gu, Jun Cheng 0003, Huazhu Fu, Kang Zhou 0001, Huaying Hao, Yitian Zhao, Tianyang Miller, Shenghua Gao, Jiang Liu 0001
IEEE Trans. Medical Imaging6
2018 Logistic Regression of Point Matches for Accurate Transformation Estimation
abstract
Feature extraction and matching (FEM) has been widely used for the registration of partially overlapping 3D shapes. Due to various factors such as imaging noise, simple geometry, or clutter, it usually introduces false positive ones. To reliably estimate the underlying transformation that brings one partial shape into the best possible alignment with another, it is critical to estimate the extent to which the established point matches are correct. To this end, we propose to use the logit function for the regression of the errors of these point matches. The novel method includes three steps: (i) normalization of the errors of the point matches, (ii) logistic regression of the point matches for the estimation of their reliabilities/weights, and (iii) estimation of the underlying transformation in the weighted least squares sense. These steps are repeated until either the maximum number of iterations has been reached or the weighted average of the errors of the point matches has been below the scanning resolution. A comparative study using real data captured by different range sensors shows that the proposed method outperforms two state-of-the-art ones for more accurate estimation of the underlying transformation.
Yonghuai Liu, Yitian Zhao, Yanquan Zhou, Jiwan Han, Wanneng Yang, Yiguang Liu
3DV2
2018 Retinal Artery and Vein Classification via Dominant Sets Clustering-Based Vascular Topology Estimation
Yitian Zhao, Jianyang Xie, Pan Su 0001, Yalin Zheng, Yonghuai Liu, Jun Cheng 0003, Jiang Liu 0001
MICCAI (2)1
2018 Uniqueness-Driven Saliency Analysis for Automated Lesion Detection with Applications to Retinal Diseases
Yitian Zhao, Yalin Zheng, Yifan Zhao 0001, Yonghuai Liu, Peng Liu 0049, Jiang Liu 0001
MICCAI (2)1
2018 Automatic 2-D/3-D Vessel Enhancement in Multiple Modality Images Using a Weighted Symmetry Filter
abstract
Automated detection of vascular structures is of great importance in understanding the mechanism, diagnosis, and treatment of many vascular pathologies. However, automatic vascular detection continues to be an open issue because of difficulties posed by multiple factors, such as poor contrast, inhomogeneous backgrounds, anatomical variations, and the presence of noise during image acquisition. In this paper, we propose a novel 2-D/3-D symmetry filter to tackle these challenging issues for enhancing vessels from different imaging modalities. The proposed filter not only considers local phase features by using a quadrature filter to distinguish between lines and edges, but also uses the weighted geometric mean of the blurred and shifted responses of the quadrature filter, which allows more tolerance of vessels with irregular appearance. As a result, this filter shows a strong response to the vascular features under typical imaging conditions. Results based on eight publicly available datasets (six 2-D data sets, one 3-D data set, and one 3-D synthetic data set) demonstrate its superior performance to other state-of-the-art methods.
Yitian Zhao, Yalin Zheng, Yonghuai Liu, Yifan Zhao 0001, Lingling Luo, Tong Na, Yongtian Wang, Jiang Liu 0001
IEEE Trans. Medical Imaging1
2017 Fuzzy rough feature selection based on OWA aggregation of fuzzy relations
abstract
The interaction between features, or attributes, of a dataset forms a major topic in machine learning and data mining. In particular, a wide range of methods have been established for feature selection, ranking, and grouping. Amongst these, fuzzy rough set based feature selection (FRFS) has been shown to be highly effective at reducing dimensionality for real-valued datasets while retaining attribute semantics. In fuzzy rough sets, the concept of crisp equivalence classes is extended by fuzzy similarity relations, and real-valued similarity measures can be captured between data instances in terms of their attribute values. Therefore, it is desirable to study the aggregation of fuzzy similarity relations to reflect the interactions between attributes. This paper presents an approach that employs OWA aggregation of fuzzy similarity relations to better perform FRFS. A high degree of modelling flexibility is provided by choosing the stress function in OWA. Experimental studies demonstrate that through using different stress functions, different features may be selected; and that given an appropriate stress function, the quality of selected features can improve over that achievable by the state-of-the-art FRFS, in performing classification tasks.
Pan Su 0001, Changjing Shang, Yitian Zhao, Tianhua Chen, Qiang Shen 0001
FUZZ-IEEE3
2017 Saliency driven vasculature segmentation with infinite perimeter active contour model
Yitian Zhao, Jingliang Zhao, Jian Yang 0009, Yonghuai Liu, Yifan Zhao 0001, Yalin Zheng, Likun Xia, Yongtian Wang
Neurocomputing1
2017 Intensity and Compactness Enabled Saliency Estimation for Leakage Detection in Diabetic and Malarial Retinopathy
abstract
Leakage in retinal angiography currently is a key feature for confirming the activities of lesions in the management of a wide range of retinal diseases, such as diabetic maculopathy and paediatric malarial retinopathy. This paper proposes a new saliency-based method for the detection of leakage in fluorescein angiography. A superpixel approach is firstly employed to divide the image into meaningful patches (or superpixels) at different levels. Two saliency cues, intensity and compactness, are then proposed for the estimation of the saliency map of each individual superpixel at each level. The saliency maps at different levels over the same cues are fused using an averaging operator. The two saliency maps over different cues are fused using a pixel-wise multiplication operator. Leaking regions are finally detected by thresholding the saliency map followed by a graph-cut segmentation. The proposed method has been validated using the only two publicly available datasets: one for malarial retinopathy and the other for diabetic retinopathy. The experimental results show that it outperforms one of the latest competitors and performs as well as a human expert for leakage detection and outperforms several state-of-the-art methods for saliency detection.
Yitian Zhao, Yalin Zheng, Yonghuai Liu, Jian Yang 0009, Yifan Zhao 0001, Duanduan Chen, Yongtian Wang
IEEE Trans. Medical Imaging1
2017 Convex Hull Aided Registration Method (CHARM)
abstract
Non-rigid registration finds many applications such as photogrammetry, motion tracking, model retrieval, and object recognition. In this paper we propose a novel convex hull aided registration method (CHARM) to match two point sets subject to a non-rigid transformation. First, two convex hulls are extracted from the source and target respectively. Then, all points of the point sets are projected onto the reference plane through each triangular facet of the hulls. From these projections, invariant features are extracted and matched optimally. The matched feature point pairs are mapped back onto the triangular facets of the convex hulls to remove outliers that are outside any relevant triangular facet. The rigid transformation from the source to the target is robustly estimated by the random sample consensus (RANSAC) scheme through minimizing the distance between the matched feature point pairs. Finally, these feature points are utilized as the control points to achieve non-rigid deformation in the form of thin-plate spline of the entire source point set towards the target one. The experimental results based on both synthetic and real data show that the proposed algorithm outperforms several state-of-the-art ones with respect to sampling, rotational angle, and data noise. In addition, the proposed CHARM algorithm also shows higher computational efficiency compared to these methods.
Jingfan Fan, Jian Yang 0009, Yitian Zhao, Danni Ai, Yonghuai Liu, Ge Wang 0001, Yongtian Wang
IEEE Trans. Vis. Comput. Graph.3
2016 An edge detection method using outer Totalistic Cellular Automata
Sebastian Amrogowicz, Yitian Zhao, Yifan Zhao 0001
Neurocomputing2
2016 3-Points Convex Hull Matching (3PCHM) for fast and robust point set registration
Jingfan Fan, Jian Yang 0009, Feng Lu 0005, Danni Ai, Yitian Zhao, Yongtian Wang
Neurocomputing5
2016 Region-based saliency estimation for 3D shape analysis and understanding
Yitian Zhao, Yonghuai Liu, Baogang Wei, Jian Yang 0009, Yifan Zhao 0001, Yongtian Wang
Neurocomputing1
2016 Convex hull indexed Gaussian mixture model (CH-GMM) for 3D point set registration
Jingfan Fan, Jian Yang 0009, Danni Ai, Likun Xia, Yitian Zhao, Yongtian Wang
Pattern Recognit.5
2016 Accurately estimating rigid transformations in registration using a boosting-inspired mechanism
Yonghuai Liu, Honghai Liu 0001, Ralph R. Martin, Luigi De Dominicis, Ran Song 0001, Yitian Zhao
Pattern Recognit.6
2015 Automated Vessel Segmentation Using Infinite Perimeter Active Contour Model with Hybrid Region Information with Application to Retinal Images
abstract
Automated detection of blood vessel structures is becoming of crucial interest for better management of vascular disease. In this paper, we propose a new infinite active contour model that uses hybrid region information of the image to approach this problem. More specifically, an infinite perimeter regularizer, provided by using L(2) Lebesgue measure of the γ -neighborhood of boundaries, allows for better detection of small oscillatory (branching) structures than the traditional models based on the length of a feature's boundaries (i.e., H(1) Hausdorff measure). Moreover, for better general segmentation performance, the proposed model takes the advantage of using different types of region information, such as the combination of intensity information and local phase based enhancement map. The local phase based enhancement map is used for its superiority in preserving vessel edges while the given image intensity information will guarantee a correct feature's segmentation. We evaluate the performance of the proposed model by applying it to three public retinal image datasets (two datasets of color fundus photography and one fluorescein angiography dataset). The proposed model outperforms its competitors when compared with other widely used unsupervised and supervised methods. For example, the sensitivity (0.742), specificity (0.982) and accuracy (0.954) achieved on the DRIVE dataset are very close to those of the second observer's annotations.
Yitian Zhao, Lavdie Rada, Ke Chen 0002, Simon P. Harding, Yalin Zheng
IEEE Trans. Medical Imaging1
2013 Using Region-Based Saliency for 3D Interest Points Detection
Yitian Zhao, Yonghuai Liu
CAIP (2)1
2012 A saliency detection based method for 3D surface simplification
abstract
To accelerate the processing for the integration, registration, representation and recognition of point clouds, it is of growing necessity to simplify the surface of 3-D models. Simplification is an approach to vary the levels of visual details as appropriate, thereby improving on the overall performance of applications. This paper proposes a saliency detection based points sampling method for mesh simplification. By generating and enhancing the saliency map, the regions which are visually important can be located. For the mesh simplification, the local details are captured by the saliency, while for the overall shape, the approach voxelizes the model and samples points in terms of the entropy of the shape index of vertices in voxels. We present a number of results to show that the method significantly simplifies the surface without distortion and loss of local details.
Yitian Zhao, Yonghuai Liu, Ran Song 0001
ICASSP1
2012 Conditional random field-based mesh saliency
abstract
We propose a new method for detecting mesh saliency, a reflection of perception-based regional importance for 3D meshes. The basic idea is to incorporate the Conditional Random Field (CRF) framework with a saliency detection process. We first produce a multi-scale representation for a mesh. Then, a CRF is designed to robustly detect salient regions utilising neighbourhood consistency. By inferring the CRF via belief propagation algorithm, we actually make use of the global statistic information in the saliency detection process. Experimental results demonstrate the robustness and the effectiveness of the proposed method.
Ran Song 0001, Yonghuai Liu, Yitian Zhao, Ralph R. Martin, Paul L. Rosin
ICIP3
2012 Extended non-local means filter for surface saliency detection
abstract
Mesh surface saliency detection is an important preprocessing step for many 3D applications. The salient region can be used to find the objects that are important on 3D surface, the benefits of saliency detection in the 3D domain include mesh simplification, registration, segmentation, compression, etc. This paper proposes a novel saliency detection method by diffusing the shape index field with non-local means filer, generating a random centre surround operator to yield saliency map and enhancing the saliency with the Retinex theory. The effectiveness of this method is demonstrated by simplification and registration. Experimental results demonstrate that the proposed approach has achieved competitive results.
Yitian Zhao, Yonghuai Liu, Ran Song 0001
ICIP1
2012 Patch based saliency detection method for 3D surface simplification
Yitian Zhao, Yonghuai Liu
ICPR1
2011 MRF-based automatic image ordering and its application to mosaicing
abstract
A fast and robust auto-sorting method for image ordering based on Markov Random Fields (MRF) is proposed. We present a specific MRF model for the ordering problem and use pairwise phase correlation for the formulation. The MRF is inferred by a modified belief propagation (BP) method. Experimental results prove that the new method can reorder a disorganised collection of images without human input, prior information or restrictions, as just the first stage of a multi stage mosaicing process, but also provides information that can be used to guide a mosaicing process in order to reduce both local mismatch and global error accumulation.
Ran Song 0001, Yonghuai Liu, Yitian Zhao, Ralph R. Martin, Paul L. Rosin
ICASSP3