EDBT 2026 Demo / reviewers in the wild / expert
ZongYuan Ge
dblp:147/2757 · also Zongyuan Ge
· DBLP profile ↗
132ranked-venue papers
6as first author
105since 2021 · last 2026
0000-0002-5880-8673ORCID · verified
Domains — the database's venue-derived domains; a paper can count in several
Graphics, computer vision, multimedia, augmented reality and games · 70 · 6 first-author · 54 since 2021Applied, interdisciplinary, general and emerging computing · 60 · 1 first-author · 52 since 2021Artificial intelligence and machine learning · 55 · 1 first-author · 45 since 2021Systems, architecture and hardware · 4 · 2 since 2021Databases, data management, data science and information retrieval · 3 · 2 since 2021Computer networks · 2 · 1 since 2021
| Year | Publication | Venue | Position |
|---|---|---|---|
| 2026 | Tears or Cheers? Benchmarking LLMs via Culturally Elicited Distinct Affective ResponsesabstractChongyuan Dai, Yaling Shen, Zihan Gao, Jia Li, Yishun Jiang, Yaxiong Wang, Liu Liu, Zongyuan Ge, Jinpeng Hu. Proceedings of the 64th Annual Meeting of the Association for Computational Linguistics (Volume 1: Long Papers). 2026. Chongyuan Dai, Yaling Shen, Jia Li 0057, Yishun Jiang, Yaxiong Wang, ZongYuan Ge, Jinpeng Hu |
ACL (1) | 8 |
| 2026 | Interictal Epileptiform Discharge Detection Through Probabilistic Diffusion Models with Maximization of Precision Recall MetricsabstractRecently, automated Interictal Epileptiform Discharge (IED) detection has attracted significant attention as a challenging predictive data analysis task aimed at improving early epilepsy diagnosis. Automated IED detection simplifies visual inspection and assists clinicians in identifying crucial IED waveform patterns in Electroencephalographic (EEG) brain activities. However, IEDs are vastly outnumbered by non-IED or background data, directly training on such data leads to a detrimental impact on model performance. Moreover, most existing methods lack high precision when tested on cross-institution datasets and this will lead to time wasted by the neurologist having to look at predicted IEDs that are not IEDs. To address these issues, we propose a novel approach that employs probabilistic diffusion models for data augmentation alongside Area Under the Precision–Recall Curve (AUPRC) maximization methods. This approach effectively addresses data scarcity and imbalance by combining real and synthesized IEDs to fully balance the training dataset, resulting in a 4.5% increase in precision and a 0.7% improvement in the F1-score during within-data evaluation. Additionally, it proves to be robust and generalizable, as evidenced by a 40.04% and 18.74% enhancement in precision and F1-score when applied to cross-data evaluation using data from another hospital. Lantian Zhang, Duong Nhu, Yun Zhao 0004, Emma Foster, Lyn Millist, Shobi Sivathamboo, Patrick Kwan, ZongYuan Ge, Lan Du 0002, Levin Kuhlmann |
Int. J. Neural Syst. | 8 |
| 2026 | Fundus image-based glaucoma screening via retinal knowledge-oriented dynamic multi-level feature integration
Chi Liu 0002, Yuzhuo Zhou, Sheng Shen 0005, ZongYuan Ge, Fengshi Jing, Shiran Zhang, Anli Wang, Feilong Yang, Tianqing Zhu, Xiaotong Han |
Knowl. Based Syst. | 4 |
| 2026 | Medical hierarchical image classification via dual-geometry image-text learningabstractHierarchical image classification is a fundamental challenge in medical image analysis, as tree-structured taxonomies inherently reflect biological and clinical relationships, spanning the general categorisation of disease entities and fine-grained cellular distinctions. Existing approaches primarily rely on multi-task learning and fine-grained detection, often requiring intricate model design and complex training strategies. In this paper, we aim to exploit the negative curvature property of hyperbolic space, which allows efficient representation of hierarchical structures. We propose a dual-geometry image-text framework, termed H 2 CL. Specifically, we introduce a lightweight classifier head on top of image backbones to extract both Euclidean and hyperbolic features, which are then combined to simultaneously preserve taxonomic consistency from an etiological perspective and enhance instance discrimination from a morphological perspective. Furthermore, a text branch is incorporated to integrate label semantics, where an entailment loss is employed to jointly model image–text alignment and inter-sample relationships. Extensive experiments on cervical cell, skin lesion, and gallbladder disease datasets demonstrate that our framework consistently outperforms advanced methods. Compared to the standard Swin Transformer, H 2 CL achieves an average accuracy improvement of 7% across all three datasets at the fine-grained level, with similarly consistent gains observed when integrated with other backbone models. The source code is publicly available at https://github.com/MCPathology/H2CL . Lei Fan 0007, Arcot Sowmya, Erik Meijering, ZongYuan Ge, Yang Song 0001 |
Medical Image Anal. | 5 |
| 2026 | Seeing cracks in frequency: FD-Mamba accurately segments cracks via frequency-difference priors
Wanqiang Cai, Junwen Zheng, Jiasong Wu, ZongYuan Ge, Laurent D. Cohen |
Pattern Recognit. | 8 |
| 2026 | Benchmarking real-world medical image classification with noisy labels: Challenges, practice, and outlookabstractLearning from noisy labels remains a major challenge in medical image analysis, where annotation demands expert knowledge and substantial inter-observer variability often leads to inconsistent or erroneous labels. Despite extensive research on learning with noisy labels (LNL), the robustness of existing methods in medical imaging has not been systematically assessed. To address this gap, we introduce LNMBench, a comprehensive benchmark for Label Noise in Medical imaging. LNMBench encompasses \textbf{10} representative methods evaluated across 7 datasets, 6 imaging modalities, and 3 noise patterns, establishing a unified and reproducible framework for robustness evaluation under realistic conditions. Comprehensive experiments reveal that the performance of existing LNL methods degrades substantially under high and real-world noise, highlighting the persistent challenges of class imbalance and domain variability in medical data. Motivated by these findings, we further propose a simple yet effective improvement to enhance model robustness under such conditions. The LNMBench codebase is publicly released to facilitate standardized evaluation, promote reproducible research, and provide practical insights for developing noise-resilient algorithms in both research and real-world medical applications.The codebase is publicly available on https://github.com/myyy777/LNMBench. Junlin Hou, Chao Zhang 0030, ZongYuan Ge, Haoran Xie 0002, Lie Ju |
Pattern Recognit. | 5 |
| 2026 | Leveraging Image-Text Pairs for Generalized Category Discovery in Medical Image ClassificationabstractGeneralized category discovery aims to identify known medical categories and unknown new medical categories from unlabeled data by migrating knowledge from labeled datasets containing only known categories, which is crucial for disease understanding and precision medicine. Many methods have been proposed and significantly improved the performance of GCD in medical images. However, most of the existing methods discover new categories based on image modalities only, ignoring useful information in the large amount of textual data related to diseases. In this paper, we propose M3GCD (Medical Multi-Modal Generalized Category Discovery), which exploits image– text pairs to jointly recognize known classes and discover novel categories in medical images. To address the varying contribution of different modalities across samples, we develop a Dynamic Expert Fusion module to automatically learn sample-specific modality weights, and further design a Local Experts Balancing mechanism to preserve the discriminative power of individual modalities. By integrating global and local perspectives, our framework adaptively balances modality contributions and enhances multi-modal robustness. Subsequently, to enable the discovery of novel unknown categories during training, we propose a Category Diffusion module grounded in the Metropolis– Hastings framework. This module adaptively merges and splits categories, allowing the model to simultaneously recognize known classes and uncover previously unseen categories during training, without requiring any prior knowledge about the unknown categories. Extensive experiments on two public multi-modal datasets (MIMIC-CXR and PatchGastric), together with a private multi-modal fundus dataset, MM-Retina, demonstrate that our method consistently improves clustering performance on both known and unknown categories compared with existing approaches. Wei Feng 0015, Sijin Zhou, ZongYuan Ge |
IEEE Trans. Medical Imaging | 5 |
| 2026 | Uncertainty-Aware Information Pursuit for Interpretable and Reliable Medical Image AnalysisabstractTo be adopted in safety-critical domains like medical image analysis, AI systems must provide human-interpretable decisions. Variational Information Pursuit (VIP) offers an interpretable-by-design framework by sequentially querying input images for human-understandable concepts, using their presence or absence to make predictions. However, existing V-IP methods overlook sample-specific uncertainty in concept predictions, which can arise from ambiguous features or model limitations, leading to suboptimal query selection and reduced robustness. In this paper, we propose an interpretable and uncertainty-aware framework for medical imaging that addresses these limitations by accounting for upstream uncertainties in concept-based, interpretable-by-design models. Specifically, we introduce two uncertainty-aware models, EUAV-IP and IUA-VIP, that integrate uncertainty estimates into the V-IP querying process to prioritize more reliable concepts per sample. EUAV-IP skips uncertain concepts via masking, while IUAV-IP incorporates uncertainty into query selection implicitly for more informed and clinically aligned decisions. Our approach allows models to make reliable decisions based on a subset of concepts tailored to each individual sample, without human intervention, while maintaining overall interpretability. We evaluate our methods on five medical imaging datasets across four modalities: dermoscopy, X-ray, ultrasound, and blood cell imaging. The proposed IUAV-IP model achieves state-of-the-art accuracy among interpretable-by-design approaches on four of the five datasets, and generates more concise explanations by selecting fewer yet more informative concepts. These advances enable more reliable and clinically meaningful outcomes, enhancing model trustworthiness and supporting safer AI deployment in healthcare. Our code and models are available at: https://github.com/Nahiduzzaman09/ UAV-IP. Md. Nahiduzzaman, Steven Korevaar, ZongYuan Ge, Feng Xia 0001, Alireza Bab-Hadiashar, Ruwan B. Tennakoon |
IEEE Trans. Medical Imaging | 3 |
| 2025 | Towards Realistic Semi-supervised Medical Image ClassificationabstractExisting semi-supervised learning (SSL) approaches follow the idealized closed-world assumption, neglecting the challenges present in realistic medical scenarios, such as open-set distribution and imbalanced class distribution. Although some methods in natural domains attempt to address the open-set problem, they are insufficient for medical domains, where intertwined challenges like class imbalance and small inter-class lesion discrepancies persist. Thus, this paper presents a novel self-recalibrated semantic training framework, which is tailored for SSL in medical imaging by ingeniously harvesting realistic unlabeled samples. Inspired by the observation that certain open-set samples share some similar disease-related representations with in-distribution samples, we first propose an informative sample selection strategy that identifies high-value samples to serve as augmentations, thereby effectively enriching the semantics of known categories. Furthermore, we adopt a compact semantic clustering strategy to address the semantic confusion raised by the above newly introduced open-set semantics. Moreover, to mitigate the interference of class imbalance in open-set SSL, we introduce a less biased dual-balanced classifier with similarity pseudo-label regularization and category-customized regularization. Extensive experiments on a variety of medical image datasets demonstrate the superior performance of our proposed method over state-of-the-art Closed-set and Open-set SSL methods. Wenxue Li 0003, Lie Ju, Peng Xia 0005, Xinyu Xiong, Lei Zhu 0002, ZongYuan Ge |
AAAI | 8 |
| 2025 | Neighbor Does Matter: Density-Aware Contrastive Learning for Medical Semi-supervised SegmentationabstractIn medical image analysis, multi-organ semi-supervised segmentation faces challenges such as insufficient labels and low contrast in soft tissues. To address these issues, existing studies typically employ semi-supervised segmentation techniques using pseudo-labeling and consistency regularization. However, these methods mainly rely on individual data samples for training, ignoring the rich neighborhood information present in the feature space. In this work, we argue that supervisory information can be directly extracted from the geometry of the feature space. Inspired by the density-based clustering hypothesis, we propose using feature density to locate sparse regions within feature clusters. Our goal is to increase intra-class compactness by addressing sparsity issues. To achieve this, we propose a Density-Aware Contrastive Learning (DACL) strategy, pushing anchored features in sparse regions towards cluster centers approximated by high-density positive samples, resulting in more compact clusters. Specifically, our method constructs density-aware neighbor graphs using labeled and unlabeled data samples to estimate feature density and locate sparse regions. We also combine label-guided co-training with density-guided geometric regularization to form complementary supervision for unlabeled data. Experiments on the Multi-Organ Segmentation Challenge dataset demonstrate that our proposed method outperforms state-of-the-art methods, highlighting its efficacy in medical image segmentation tasks. Zhongxing Xu, Wenxue Li 0003, Peng Xia 0005, Yiheng Zhong, Hanjun Wu, Jionglong Su, ZongYuan Ge |
AAAI | 9 |
| 2025 | Toward Modality Gap: Vision Prototype Learning for Weakly-supervised Semantic Segmentation with CLIPabstractThe application of Contrastive Language-Image Pre-training (CLIP) in Weakly Supervised Semantic Segmentation (WSSS) research demonstrates powerful cross-modal semantic understanding capabilities. Existing methods attempt to optimize input text prompts for improved alignment of images and text, specifically by finely adjusting text prototypes to facilitate semantic matching. Nevertheless, given the modality gap between text and vision spaces, the text prototypes employed by these methods have not effectively established a close correspondence with pixel-level vision features. In this work, our theoretical analysis indicates that the inherent modality gap results in misalignment of text and region features, and that this gap cannot be sufficiently reduced by minimizing contrast loss in CLIP. To mitigate the impact of the modality gap, we propose a Vision Prototype Learning (VPL) framework, by introducing more representative vision prototypes. The core of this framework is to learn class-specific vision prototypes in vision space with the help of text prototypes, to capture high-quality localization maps. Moreover, we propose a regional semantic contrast module that contrasts regions embedding with corresponding prototypes, leading to more comprehensive and robust feature learning. Experimental results show that our proposed framework achieves state-of-the-art performance on two benchmark datasets. Zhongxing Xu, Yingxue Su, Zhiyi Zhao, Jionglong Su, ZongYuan Ge |
AAAI | 8 |
| 2025 | Enhancing Interpretable Image Classification Through LLM Agents and Conditional Concept Bottleneck ModelsabstractConcept Bottleneck Models (CBMs) decompose image classification into a process governed by interpretable, human-readable concepts. Recent advances in CBMs have used Large Language Models (LLMs) to generate candidate concepts. However, a critical question remains: What is the optimal number of concepts to use? Current concept banks suffer from redundancy or insufficient coverage. To address this issue, we introduce a dynamic, agent-based approach that adjusts the concept bank in response to environmental feedback, optimizing the number of concepts for sufficiency yet concise coverage. Moreover, we propose Conditional Concept Bottleneck Models (CoCoBMs) to overcome the limitations in traditional CBMs’ concept scoring mechanisms. It enhances the accuracy of assessing each concept’s contribution to classification tasks and feature an editable matrix that allows LLMs to correct concept scores that conflict with their internal knowledge. Our evaluations across 6 datasets show that our method not only improves classification accuracy by 6% but also enhances interpretability assessments by 30%. Yiwen Jiang, Deval Mehta 0001, Wei Feng 0015, ZongYuan Ge |
ACL (1) | 4 |
| 2025 | MMRC: A Large-Scale Benchmark for Understanding Multimodal Large Language Model in Real-World ConversationabstractRecent multimodal large language models (MLLMs) have demonstrated significant potential in open-ended conversation, generating more accurate and personalized responses. However, their abilities to memorize, recall, and reason in sustained interactions within real-world scenarios remain underexplored. This paper introduces MMRC, a Multi-Modal Real-world Conversation benchmark for evaluating six core open-ended abilities of MLLMs: information extraction, multi-turn reasoning, information update, image management, memory recall, and answer refusal. With data collected from real-world scenarios, MMRC comprises 5,120 conversations and 28,720 corresponding manually labeled questions, posing a significant challenge to existing MLLMs. Evaluations on 20 MLLMs in MMRC indicate an accuracy drop during open-ended interactions. We identify four common failure patterns: long-term memory degradation, inadequacies in updating factual knowledge, accumulated assumption of error propagation, and reluctance to “say no.” To mitigate these issues, we propose a simple yet effective NOTE-TAKING strategy, which can record key information from the conversation and remind the model during its responses, enhancing conversational capabilities. Experiments across six MLLMs demonstrate significant performance improvements. Haochen Xue, Yexin Liu, Qidong Huang, Yulong Li 0002, Zhongxing Xu, Chong Zhang 0006, Yutong Xie 0001, Muhammad Imran Razzak, ZongYuan Ge, Jionglong Su, Junjun He, Yu Qiao 0001 |
ACL (1) | 13 |
| 2025 | PG-SAM: A Fine-Grained Prior-Guided SAM Framework for Prompt-Free Medical Image SegmentationabstractSegment Anything Model (SAM) demonstrates powerful zero-shot capabilities; however, its accuracy and robustness significantly decrease when applied to medical image segmentation. Existing methods address this issue through modality fusion, integrating textual and image information to provide more detailed priors. In this study, we argue that the granularity of text and the domain gap affect the accuracy of the priors. Furthermore, the discrepancy between high-level abstract semantics and pixel-level boundary details in images can introduce noise into the fusion process. To address this, we propose Prior-Guided SAM (PG-SAM), which employs a fine-grained modality prior aligner to leverage specialized medical knowledge for better modality alignment. The core of our method lies in efficiently addressing the domain gap with fine-grained text from a medical large language model (LLM). Meanwhile, it also enhances the priors' quality after modality alignment, ensuring more accurate segmentation. In addition, our decoder enhances the model's expressive capabilities through multi-level feature fusion and iterative mask optimizer operations, supporting unprompted learning. We also propose a unified pipeline that effectively supplies high-quality semantic information to SAM. Extensive experiments on the datasets demonstrate that the proposed PG-SAM achieves state-of-the-art performance. Our anonymous code is released at https://github.com/logan-0623/PG-SAM. Yiheng Zhong, Zihong Luo, Yingzhen Hu, Zelin Peng, Jionglong Su, ZongYuan Ge, Muhammad Imran Razzak |
BIBM | 9 |
| 2025 | HGCLIP: Exploring Vision-Language Models with Graph Representations for Hierarchical UnderstandingabstractObject categories are typically organized into a multi-granularity taxonomic hierarchy. When classifying categories at different hierarchy levels, traditional uni-modal approaches focus primarily on image features, revealing limitations in complex scenarios. Recent studies integrating Vision-Language Models (VLMs) with class hierarchies have shown promise, yet they fall short of fully exploiting the hierarchical relationships. These efforts are constrained by their inability to perform effectively across varied granularity of categories. To tackle this issue, we propose a novel framework (HGCLIP) that effectively combines CLIP with a deeper exploitation of the Hierarchical class structure via Graph representation learning. We explore constructing the class hierarchy into a graph, with its nodes representing the textual or image features of each category. After passing through a graph encoder, the textual features incorporate hierarchical structure information, while the image features emphasize class-aware features derived from prototypes through the attention mechanism. Our approach demonstrates significant improvements on 11 diverse visual recognition benchmarks. Our codes are fully available at https://github.com/richard-peng-xia/HGCLIP. Peng Xia 0005, Xingtong Yu, Lie Ju, Zhiyong Wang 0001, Peibo Duan, ZongYuan Ge |
COLING | 7 |
| 2025 | Seeing Far and Clearly: Mitigating Hallucinations in MLLMs with Attention Causal DecodingabstractRecent advancements in multimodal large language models (MLLMs) have significantly improved performance in visual question answering. However, they often suffer from hallucinations. In this work, hallucinations are categorized into two main types: initial hallucinations and snowball hallucinations. We argue that adequate contextual information can be extracted directly from the token interaction process. Inspired by causal inference in the decoding strategy, we propose to leverage causal masks to establish information propagation between multimodal tokens. The hypothesis is that insufficient interaction between those tokens may lead the model to rely on outlier tokens, overlooking dense and rich contextual cues. Therefore, we propose to intervene in the propagation process by tackling outlier tokens to enhance in-context inference. With this goal, we present FarSight, a versatile plug-and-play decoding strategy to reduce attention interference from outlier tokens merely by optimizing the causal mask. The heart of our method is effective token propagation. We design an attention register structure within the upper triangular matrix of the causal mask, dynamically allocating attention to capture attention diverted to outlier tokens. Moreover, a positional awareness encoding method with a diminishing masking rate is proposed, allowing the model to attend to further preceding tokens, especially for video sequence tasks. With extensive experiments, FarSight demonstrates significant hallucination-mitigating performance across different MLLMs on both image and video benchmarks, proving its effectiveness. Zhongxing Xu, Zile Huang, Haochen Xue, Ziyang Chen 0003, Zelin Peng, Sijin Zhou, Wenxue Li 0003, Yulong Li 0002, Wenxuan Song, Shiyan Su, Wei Feng 0015, Jionglong Su, Mingquan Lin, Yifan Peng 0002, Xuelian Cheng, Muhammad Imran Razzak, ZongYuan Ge |
CVPR | 21 |
| 2025 | WISE: Weak-Supervision-Guided Step-by-Step Explanations for Multimodal LLMs in Image ClassificationabstractMultimodal Large Language Models (MLLMs) have shown promise in visual-textual reasoning, with Multimodal Chain-of-Thought (MCoT) prompting significantly enhancing interpretability.However, existing MCoT methods rely on rationale-rich datasets and largely focus on inter-object reasoning, overlooking the intraobject understanding crucial for image classification.To address this gap, we propose WISE, a Weak-supervIsion-guided Step-bystep Explanation method that augments any image classification dataset with MCoTs by reformulating the concept-based representations from Concept Bottleneck Models (CBMs) into concise, interpretable reasoning chains under weak supervision.Experiments across ten datasets show that our generated MCoTs not only improve interpretability by 37% but also lead to gains in classification accuracy when used to fine-tune MLLMs 1 .Our work bridges concept-based interpretability and generative MCoT reasoning, providing a generalizable framework for enhancing MLLMs in fine-grained visual understanding. Yiwen Jiang, Deval Mehta 0001, Siyuan Yan, Yaling Shen, ZongYuan Ge |
EMNLP | 6 |
| 2025 | SynFER: Towards Boosting Facial Expression Recognition With Synthetic DataabstractFacial expression datasets remain limited in scale due to the subjectivity of annotations and the labor-intensive nature of data collection. This limitation poses a significant challenge for developing modern deep learning-based facial expression analysis models, particularly foundation models, that rely on large-scale data for optimal performance. To tackle the overarching and complex challenge, instead of introducing a new large-scale dataset, we introduce SynFER (Synthesis of Facial Expressions with Refined Control), a novel synthetic framework for synthesizing facial expression image data based on high-level textual descriptions as well as more fine-grained and precise control through facial action units. To ensure the quality and reliability of the synthetic data, we propose a semantic guidance technique to steer the generation process and a pseudo-label generator to help rectify the facial expression labels for the synthetic images. To demonstrate the generation fidelity and the effectiveness of the synthetic data from SynFER, we conduct extensive experiments on representation learning using both synthetic data and real-world data. Results validate the efficacy of our approach and the synthetic data. Notably, our approach achieves a 67.23% classification accuracy on AffectNet when training solely with synthetic data equivalent to the AffectNet training set size, which increases to 69.84% when scaling up to five times the original size. Code is available here. Xilin He, Xiaole Xian, Bing Li 0024, Muhammad Haris Khan, ZongYuan Ge, Weicheng Xie 0001, Siyang Song, LinLin Shen, Bernard Ghanem, Xiangyu Yue 0001 |
ICCV | 6 |
| 2025 | OphCLIP: Hierarchical Retrieval-Augmented Learning for Ophthalmic Surgical Video-Language PretrainingabstractSurgical practice involves complex visual interpretation, procedural skills, and advanced medical knowledge, making surgical vision-language pretraining (VLP) particularly challenging due to this complexity and the limited availability of annotated data. To address the gap, we propose OphCLIP, a hierarchical retrieval-augmented vision-language pretraining framework specifically designed for ophthalmic surgical workflow understanding. OphCLIP leverages the OphVL dataset we constructed, a large-scale and comprehensive collection of over 375K hierarchically structured video-text pairs with tens of thousands of different combinations of attributes (surgeries, phases/operations/actions, instruments, medications, as well as more advanced aspects like the causes of eye diseases, surgical objectives, and postoperative recovery recommendations, etc). These hierarchical video-text correspondences enable OphCLIP to learn both fine-grained and long-term visual representations by aligning short video clips with detailed narrative descriptions and full videos with structured titles, capturing intricate surgical details and high-level procedural insights, respectively. Our OphCLIP also designs a retrieval-augmented pretraining framework to leverage the underexplored large-scale silent surgical procedure videos, automatically retrieving semantically relevant content to enhance the representation learning of narrative videos. Evaluation across 11 datasets for phase recognition and multi-instrument identification shows OphCLIP's robust generalization and superior performance. Kun Yuan 0004, Yaling Shen, Xiaohao Xu, Wei Li 0320, Zhongxing Xu, Zelin Peng, Siyuan Yan, Vinkle Srivastav, Diping Song, Tianbin Li, Danli Shi, Jin Ye 0002, Nicolas Padoy, Nassir Navab, Junjun He, ZongYuan Ge |
ICCV | 20 |
| 2025 | Derm1M: A Million-Scale Vision-Language Dataset Aligned with Clinical Ontology Knowledge for DermatologyabstractThe emergence of vision-language models has transformed medical AI, enabling unprecedented advances in diagnostic capability and clinical applications. However, progress in dermatology has lagged behind other medical domains due to the lack of standard image-text pairs. Existing dermatological datasets are limited in both scale and depth, offering only single-label annotations across a narrow range of diseases instead of rich textual descriptions, and lacking the crucial clinical context needed for real-world applications. To address these limitations, we present Derm1M, the first large-scale vision-language dataset for dermatology, comprising 1,029,761 image-text pairs. Built from diverse educational resources and structured around a standard ontology collaboratively developed by experts, Derm1M provides comprehensive coverage for over 390 skin conditions across four hierarchical levels and 130 clinical concepts with rich contextual information such as medical history, symptoms, and skin tone. To demonstrate Derm1M potential in advancing both AI research and clinical application, we pretrained a series of CLIP-like models, collectively called DermLIP, on this dataset. The DermLIP family significantly outperforms state-of-the-art foundation models on eight diverse datasets across multiple tasks, including zero-shot skin disease classification, clinical and artifacts concept identification, few-shot/full-shot learning, and cross-modal retrieval. Our dataset and code will be publicly available at https://github.com/SiyuanYan1/Derm1M upon acceptance. Siyuan Yan, Yiwen Jiang, Xieji Li, Hao Fei 0001, Philipp Tschandl, Harald Kittler, ZongYuan Ge |
ICCV | 8 |
| 2025 | MoRE: Unlocking Scalability in Reinforcement Learning for Quadruped Vision-Language-Action ModelsabstractDeveloping versatile quadruped robots that can smoothly perform various actions and tasks in real-world environments remains a significant challenge. This paper introduces a novel vision-language-action (VLA) model, mixture of robotic experts (MoRE), for quadruped robots that aim to introduce reinforcement learning (RL) for fine-tuning large-scale VLA models with a large amount of mixed-quality data. MoRE integrates multiple low-rank adaptation modules as distinct experts within a dense multi-modal large language model (MLLM), forming a sparse-activated mixture-of-experts model. This design enables the model to effectively adapt to a wide array of downstream tasks. Moreover, we employ a reinforcement learning-based training objective to train our model as a Q-function after deeply exploring the structural properties of our tasks. Effective learning from automatically collected mixed-quality data enhances data efficiency and model performance. Extensive experiments demonstrate that MoRE outperforms all baselines across six different skills and exhibits superior generalization capabilities in out-of-distribution scenarios. We further validate our method in real-world scenarios, confirming the practicality of our approach and laying a solid foundation for future research on multi-task learning in quadruped robots. Han Zhao 0008, Wenxuan Song, Xinyang Tong, Pengxiang Ding, Xuelian Cheng, ZongYuan Ge |
ICRA | 7 |
| 2025 | PD-VLA: Accelerating Vision-Language-Action Model Integrated with Action Chunking via Parallel DecodingabstractVision-Language-Action (VLA) models demonstrate remarkable potential for generalizable robotic manipulation. The performance of VLA models can be improved by integrating with action chunking, a critical technique for effective control. However, action chunking linearly scales up action dimensions in VLA models with increased chunking sizes. This reduces the inference efficiency. Therefore, accelerating VLA integrated with action chunking is an urgent need. To tackle this problem, we propose PD-VLA, the first parallel decoding framework for VLA models integrated with action chunking. Our framework reformulates autoregressive decoding as a nonlinear system solved by parallel fixed-point iterations. This approach preserves model performance with mathematical guarantees while significantly improving decoding speed. In addition, it enables training-free acceleration without architectural changes, as well as seamless synergy with existing acceleration techniques. Extensive simulations validate that our PD-VLA maintains competitive success rates while achieving 2.52× execution frequency on manipulators (with 7 degrees of freedom) compared with the fundamental VLA model. Furthermore, we experimentally identify the most effective settings for acceleration. Finally, real-world experiments validate its high applicability across different tasks. Wenxuan Song, Pengxiang Ding, Han Zhao 0008, Zhide Zhong, ZongYuan Ge, Jun Ma 0008, Haoang Li |
IROS | 7 |
| 2025 | Enhancing Fundus Image-Based Glaucoma Screening via Dynamic Global-Local Feature Integration
Yuzhuo Zhou, Chi Liu 0002, Sheng Shen 0005, Siyu Le, Sihan Ouyang, ZongYuan Ge |
KSEM (2) | 7 |
| 2025 | Delving Into Out-of-Distribution Detection with Medical Vision-Language Models
Lie Ju, Sijin Zhou, Huimin Lu 0001, Zhuoting Zhu, Pearse A. Keane, ZongYuan Ge |
MICCAI (5) | 7 |
| 2025 | Ophora: A Large-Scale Data-Driven Text-Guided Ophthalmic Surgical Video Generation Model
Wei Li 0320, Guoan Wang, Kaijing Zhou, Junzhi Ning, ZongYuan Ge, Lixu Gu, Junjun He |
MICCAI (9) | 8 |
| 2025 | Knowledge Tree Driven Contextualized Instruction Tuning of Foundation Models for Epilepsy Drug Recommendation
Duy Khoa Pham, Deval Mehta 0001, Yiwen Jiang, Daniel Thom, Richard Shek-kwan Chang, Mohammad Nazem-Zadeh, Emma Foster, Timothy Fazio, Sarah Holper, Karin Verspoor, Jiahe Liu, Duong Nhu, Sarah Barnard, Terence J. O'Brien, Jacqueline French, Patrick Kwan, ZongYuan Ge |
MICCAI (6) | 18 |
| 2025 | Controllable Skin Synthesis via Lesion-Focused Vector Autoregression Model
Siyuan Yan, Jason J. Ong, ZongYuan Ge, Lei Zhang 0095 |
MICCAI (16) | 5 |
| 2025 | Robust Multimodal Learning for Ophthalmic Disease Grading via Disentangled Representation
Xinkun Wang, Yifang Wang 0013, Senwei Liang, Junjun He, ZongYuan Ge, Muhammad Imran Razzak |
MICCAI (8) | 9 |
| 2025 | MSWAL: 3D Multi-class Segmentation of Whole Abdominal Lesions Dataset
Zhaodong Wu, Qiaochu Zhao, Yulong Li 0002, Haochen Xue, Zhengyong Jiang, Angelos Stefanidis, Muhammad Imran Razzak, ZongYuan Ge, Junjun He, Yu Qiao 0001, Kang Dang, Jionglong Su |
MICCAI (2) | 10 |
| 2025 | MAKE: Multi-Aspect Knowledge-Enhanced Vision-Language Pretraining for Zero-Shot Dermatological Assessment
Siyuan Yan, Xieji Li, Yiwen Jiang, ZongYuan Ge |
MICCAI (5) | 6 |
| 2025 | CrackMamba with Normalized Soft-Frangi-Filter Enhancement towards Accurate Crack SegmentationabstractCrack segmentation is crucial in monitoring infrastructure degradation. However, the strong contextual dependencies of long-spanned morphology and numerous fine-grained branches hinder high-precision segmentation performance. To tackle the above obstacles, we provide a pure Mamba-based model termed CrackMamba for accurate crack segmentation. CrackMamba utilizes the Mamba-based Feature Extractor (MFE) to effectively model global dependencies of long-spanned cracks with linear computational complexity for powerful representation. To capture fine-grained crack branches, we design a novel texture refinement module, which employs a multi-scale aggregation strategy and utilizes the MFE and a Normalized Soft-Frangi-Filter (NSFF) module to integrate hierarchical features from the decoder. The CrackMamba and NSFF exhibit strong complementarity. The NSFF module enhances the capability of CrackMamba to segment fine-grained crack textures, while CrackMamba effectively eliminating crack-unrelated curves extracted by NSFF module. Extensive experiments are conducted on two benchmark crack datasets, and the results demonstrate that the proposed CrackMamba achieves the state-of-the-art (SOTA) performance with fewer parameters and higher computational efficiency. Wanqiang Cai, Yingyao Ma, Jiasong Wu, ZongYuan Ge, Bin Wang 0041 |
ICMR | 6 |
| 2025 | ReactDiff: Fundamental Multiple Appropriate Facial Reaction Diffusion ModelabstractThe automatic generation of diverse and human-like facial reactions in dyadic dialogue remains a critical challenge for human-computer interaction systems. Existing methods fail to model the stochasticity and dynamics inherent in real human reactions. To address this, we propose ReactDiff, a novel temporal diffusion framework for generating diverse facial reactions that are appropriate for responding to any given dialogue context. Our key insight is that plausible human reactions demonstrate smoothness, and coherence over time, and conform to constraints imposed by human facial anatomy. To achieve this, ReactDiff incorporates two vital priors (spatio-temporal facial kinematics) into the diffusion process: i) temporal facial behavioral kinematics and ii) facial action unit dependencies. These two constraints guide the model toward realistic human reaction manifolds, avoiding visually unrealistic jitters, unstable transitions, unnatural expressions, and other artifacts. Extensive experiments on the REACT2024 dataset demonstrate that our approach not only achieves state-of-the-art reaction quality but also excels in diversity and reaction appropriateness. Our code is publicly available at https://github.com/lingjivoo/ReactDiff. Siyang Song, Siyuan Yan, ZongYuan Ge |
ACM Multimedia | 5 |
| 2025 | Generalized Category Discovery under Domain Shift: A Frequency Domain PerspectiveabstractGeneralized Category Discovery (GCD) aims to leverage labeled samples from known categories to cluster unlabeled data that may include both known and unknown categories. While existing methods have achieved impressive results under standard conditions, their performance often deteriorates in the presence of distribution shifts. In this paper, we explore a more realistic task: Domain-Shifted Generalized Category Discovery (DS\_GCD), where the unlabeled data includes not only unknown categories but also samples from unknown domains. To tackle this challenge, we propose a \textbf{\underline{F}}requency-guided Gene\textbf{\underline{r}}alized Cat\textbf{\underline{e}}gory Discov\textbf{\underline{e}}ry framework (FREE) that enhances the model's ability to discover categories under distributional shift by leveraging frequency-domain information. Specifically, we first propose a frequency-based domain separation strategy that partitions samples into known and unknown domains by measuring their amplitude differences. We then propose two types of frequency-domain perturbation strategies: a cross-domain strategy, which adapts to new distributions by exchanging amplitude components across domains, and an intra-domain strategy, which enhances robustness to intra-domain variations within the unknown domain. Furthermore, we extend the self-supervised contrastive objective and semantic clustering loss to better guide the training process. Finally, we introduce a clustering-difficulty-aware resampling technique to adaptively focus on harder-to-cluster categories, further enhancing model performance. Extensive experiments demonstrate that our method effectively mitigates the impact of distributional shifts across various benchmark datasets and achieves superior performance in discovering both known and unknown categories. Wei Feng 0015, ZongYuan Ge |
NeurIPS | 2 |
| 2025 | Towards Dynamic 3D Reconstruction of Hand-Instrument Interaction in Ophthalmic SurgeryabstractAccurate 3D reconstruction of hands and instruments is critical for vision-based analysis of ophthalmic microsurgery, yet progress has been hampered by the lack of realistic, large-scale datasets and reliable annotation tools. In this work, we introduce OphNet-3D, the first extensive RGB-D dynamic 3D reconstruction dataset for ophthalmic surgery, comprising 41 sequences from 40 surgeons and totaling 7.1 million frames, with fine-grained annotations of 12 surgical phases, 10 instrument categories, dense MANO hand meshes, and full 6-DoF instrument poses. To scalably produce high-fidelity labels, we design a multi-stage automatic annotation pipeline that integrates multi-view data observation, data-driven motion prior with cross-view geometric consistency and biomechanical constraints, along with a combination of collision-aware interaction constraints for instrument interactions. Building upon OphNet-3D, we establish two challenging benchmarks—bimanual hand pose estimation and hand–instrument interaction reconstruction—and propose two dedicated architectures: H-Net for dual-hand mesh recovery and OH-Net for joint reconstruction of two-hand–two-instrument interactions. These models leverage a novel spatial reasoning module with weak-perspective camera modeling and collision-aware center-based representation. Both architectures outperform existing methods by substantial margins, achieving improvements of over 2mm in Mean Per Joint Position Error (MPJPE) and up to 23\% in ADD-S metrics for hand and instrument reconstruction, respectively. Zhengdi Yu, Yulong Li 0002, Muhammad Imran Razzak, Junjun He, Tolga Birdal, Kaijing Zhou, ZongYuan Ge |
NeurIPS | 10 |
| 2025 | Decoding Causal Structure: End-to-End Mediation Pathways InferenceabstractCausal mediation analysis is crucial for deconstructing complex mechanisms of action. However, in current mediation analysis, complex structures derived from causal discovery lack direct interpretation of mediation pathways, while traditional mediation analysis and effect estimation are limited by the reliance on pre-specified pathways, leading to a disconnection between structure discovery and causal mechanism understanding. Therefore, a unified framework integrating structure discovery, pathway identification, and effect estimation systematically quantifies mediation pathways under structural uncertainty, enabling automated identification and inference of mediation pathways. To this end, we propose Structure-Informed Guided Mediation Analysis (SIGMA), which guides automated mediation pathway identification through probabilistic causal structure discovery and uncertainty quantification, enabling end-to-end propagation of structural uncertainty from structure learning to effect estimation. Specifically, SIGMA employs differentiable Flow-Structural Equation Models to learn structural posteriors, generating diverse Directed Acyclic Graphs (DAGs) to quantify structural uncertainty. Based on these DAGs, we introduce the Path Stability Score to evaluate the marginal probability of pathways, identifying high-confidence mediation paths. For identified mediation pathways, we integrate Efficient Influence Functions with Bayesian model averaging to fuse within-structure estimation uncertainty and between-structure effect variation, propagating uncertainty to the final effect estimates. In synthetic data experiments, SIGMA achieves state-of-the-art performance in pathway identification accuracy and effect quantification precision under structures uncertainty, concurrent multiple pathways, and nonlinear scenarios. In real-world applications using Human Phenotype Project data, SIGMA identifies mediation effects of sleep quality on cardiovascular health through inflammatory and metabolic pathways, uncovering previously unspecified multiple mediation paths. Yulong Li 0002, Xiwei Liu, Jionglong Su, ZongYuan Ge, Muhammad Imran Razzak, Eran Segal |
NeurIPS | 6 |
| 2025 | UniViT: Unifying Image and Video Understanding in One Vision EncoderabstractDespite the impressive progress of recent pretraining methods on multimodal tasks, existing methods are inherently biased towards either spatial modeling (e.g., CLIP) or temporal modeling (e.g., V-JEPA), limiting their joint capture of spatial details and temporal dynamics. To this end, we propose UniViT, a cluster-driven unified self-supervised learning framework that effectively captures the structured semantics of both image spatial content and video temporal dynamics through event-level and object-level clustering and discrimination. Specifically, we leverage offline clustering to generate semantic clusters across both modalities. For videos, multi-granularity event-level clustering progressively expands from single-event to structured multi-event segments, capturing coarse-to-fine temporal semantics; for images, object-level clustering captures fine-grained spatial semantics. However, while global clustering provides semantically consistent clusters, it lacks modeling of structured semantic relations (e.g., temporal event structures). To address this, we introduce a contrastive objective that leverages these semantic clusters as pseudo-label supervision to explicitly enforce structural constraints, including temporal event relations and spatial object co-occurrences, capturing structured semantics beyond categories. Meanwhile, UniViT jointly embeds structured object-level and event-level semantics into a unified representation space. Furthermore, UniViT introduces two key components: (i) Unified Rotary Position Embedding integrates relative positional embedding with frequency-aware dimension allocation to support position-invariant semantic learning and enhance the stability of structured semantics in the discrimination stage; and (ii) Variable Spatiotemporal Streams adapt to inputs of varying frame lengths, addressing the rigidity of conventional fixed-input approaches. Extensive experiments across varying model scales demonstrate that UniViT achieves state-of-the-art performance on linear probing, attentive probing, question answering, and spatial understanding tasks. Xiang An, Yin Xie, Kaicheng Yang 0002, Zimin Ran, Muhammad Imran Razzak, Ziyong Feng, Behzad Bozorgtabar, Jiankang Deng, ZongYuan Ge |
NeurIPS | 14 |
| 2025 | Local Masked Reconstruction for Efficient Self-Supervised Learning on High-Resolution ImagesabstractSelf-supervised learning for computer vision has progressed tremendously and improved many downstream vision tasks, such as image classification, semantic segmentation, and object detection. Among these, generative self-supervised vision learning approaches, such as MAE and BEiT, show promising performance. However, their global reconstruction mechanism is computationally demanding, especially for high-resolution images. The computational cost increases extensively when scaled to a large-scale dataset. To address this issue, we propose local masked reconstruction (LoMaR), a simple yet effective approach that reconstructs image patches from small neighboring regions. The strategy can be easily integrated into any generative self-supervised learning techniques and improves the trade-off between efficiency and accuracy compared to reconstruction over the entire image. LoMaR is$2.5\times faster$than MAE and 5.0x faster than BEiT on$384\times 384$ImageNet pretraining and surpasses them by 0.2% and 0.8% in accuracy, respectively. It is$2.1\times faster$than MAE on iNaturalist pretraining and gains 0.2% in accuracy. On MS COCO, LoMaR outperforms MAE by 0.5$AP^{box}$on object detection and 0.5$AP^{mask}$on instance segmentation. It also outperforms$MAE$by 0.2% on semantic segmentation. Our code and pretrained models are available at: https://github.com/junchen14/LoMaR. Jun Chen 0021, Faizan Farooq Khan, Ammar Sherif, ZongYuan Ge, Boyang Li 0001, Mohamed Elhoseiny 0001 |
WACV | 5 |
| 2025 | T-Person-GAN: Text-to-Person image generation with identity-consistency and manifold mix-up
Deyin Liu, Lin Wu 0001, Bo Li 0090, Ye Zhao 0001, ZongYuan Ge |
Expert Syst. Appl. | 5 |
| 2025 | Adaptive transformer modelling of density function for nonparametric survival analysisabstractAbstract Survival analysis holds a crucial role across diverse disciplines, such as economics, engineering and healthcare. It empowers researchers to analyze both time-invariant and time-varying data, encompassing phenomena like customer churn, material degradation and various medical outcomes. Given the complexity and heterogeneity of such data, recent endeavors have demonstrated successful integration of deep learning methodologies to address limitations in conventional statistical approaches. However, current methods typically involve cluttered probability distribution function (PDF), have lower sensitivity in censoring prediction, only model static datasets, or only rely on recurrent neural networks for dynamic modelling. In this paper, we propose a novel survival regression method capable of producing high-quality unimodal PDFs without any prior distribution assumption, by optimizing novel Margin-Mean-Variance loss and leveraging the flexibility of Transformer to handle both temporal and non-temporal data, coined UniSurv. Extensive experiments on several datasets demonstrate that UniSurv places a significantly higher emphasis on censoring compared to other methods. Deval Mehta 0001, David Darby, Daniel Merlo, Melissa Gresle, Anneke van der Walt, Helmut Butzkueven, ZongYuan Ge |
Mach. Learn. | 11 |
| 2025 | Neighbor-Guided Unbiased Framework for Generalized Category Discovery in Medical Image ClassificationabstractGeneralized category discovery (GCD) utilizes seen category knowledge to automatically discover new semantic categories that are not defined in the training phase. Nevertheless, there has been no research conducted on identifying new classes using medical images and disease categories, which is essential for understanding and diagnosing specific diseases. Moreover, existing methods still produce predictions that are biased towards seen categories since the model is mainly supervised by labeled seen categories, which in turn leads to sub-optimal clustering performance. In this paper, we propose a new neighbor-guided unbiased framework (NGUF) that leverages neighbor information to mitigate prediction bias to address the GCD problem in medical tasks. Specifically, we devise a neighbor-guided cross-pseudo-clustering strategy, which exploits the knowledge of the nearest-neighbor samples to adjust the model predictions thereby generating unbiased pseudo-clustering supervision. Then, based on the unbiased pseudo-clustering supervision, we use a view-invariant learning strategy to assign labels to all samples. In addition, we propose an adaptive weight learning strategy that dynamically determines the degree of adjustment of the predictions of different samples based on the distance density values. Finally, we further propose a cross-batch knowledge distillation module to utilize information from successive iterations to encourage training consistency. Extensive experiments on four medical image datasets show that NGUF is effective in mitigating the model's prediction bias and has superior performance to other state-of-the-art GCD algorithms. Our code will be released soon. Wei Feng 0015, Sijin Zhou, Yiwen Jiang, ZongYuan Ge |
IEEE J. Biomed. Health Informatics | 5 |
| 2025 | Multi-Label Generalized Zero Shot Chest X-Ray Classification by Combining Image-Text Information With Feature DisentanglementabstractIn fully supervised learning-based medical image classification, the robustness of a trained model is influenced by its exposure to the range of candidate disease classes. Generalized Zero Shot Learning (GZSL) aims to correctly predict seen and novel unseen classes. Current GZSL approaches have focused mostly on the single-label case. However, it is common for chest X-rays to be labelled with multiple disease classes. We propose a novel multi-modal multi-label GZSL approach that leverages feature disentanglement andmulti-modal information to synthesize features of unseen classes. Disease labels are processed through a pre-trained BioBert model to obtain text embeddings that are used to create a dictionary encoding similarity among different labels. We then use disentangled features and graph aggregation to learn a second dictionary of inter-label similarities. A subsequent clustering step helps to identify representative vectors for each class. The multi-modal multi-label dictionaries and the class representative vectors are used to guide the feature synthesis step, which is the most important component of our pipeline, for generating realistic multi-label disease samples of seen and unseen classes. Our method is benchmarked against multiple competing methods and we outperform all of them based on experiments conducted on the publicly available NIH and CheXpert chest X-ray datasets. Dwarikanath Mahapatra, Antonio Jimeno-Yepes, Behzad Bozorgtabar, Sudipta Roy 0002, ZongYuan Ge, Mauricio Reyes 0001 |
IEEE Trans. Medical Imaging | 5 |
| 2025 | Corrections to "Multi-Label Generalized Zero Shot Chest X-Ray Classification By Combining Image-Text Information With Feature Disentanglement"abstractPresents corrections to the paper, (Corrections to "Multi-Label Generalized Zero Shot Chest X-Ray Classification By Combining Image-Text Information With Feature Disentanglement"). Dwarikanath Mahapatra, Antonio Jimeno-Yepes, Behzad Bozorgtabar, Sudipta Roy 0002, ZongYuan Ge, Mauricio Reyes 0001 |
IEEE Trans. Medical Imaging | 5 |
| 2025 | Prompt-Driven Latent Domain Generalization for Medical Image ClassificationabstractDeep learning models for medical image analysis easily suffer from distribution shifts caused by dataset artifact bias, camera variations, differences in the imaging station, etc., leading to unreliable diagnoses in real-world clinical settings. Domain generalization (DG) methods, which aim to train models on multiple domains to perform well on unseen domains, offer a promising direction to solve the problem. However, existing DG methods assume domain labels of each image are available and accurate, which is typically feasible for only a limited number of medical datasets. To address these challenges, we propose a unified DG framework for medical image classification without relying on domain labels, called Prompt-driven Latent Domain Generalization (PLDG). PLDG consists of unsupervised domain discovery and prompt learning. This framework first discovers pseudo domain labels by clustering the bias-associated style features, then leverages collaborative domain prompts to guide a Vision Transformer to learn knowledge from discovered diverse domains. To facilitate cross-domain knowledge learning between different prompts, we introduce a domain prompt generator that enables knowledge sharing between domain prompts and a shared prompt. A domain mixup strategy is additionally employed for more flexible decision margins and mitigates the risk of incorrect domain assignments. Extensive experiments on three medical image classification tasks and one debiasing task demonstrate that our method can achieve comparable or even superior performance than conventional DG algorithms without relying on domain labels. Our code is publicly available at https://github.com/SiyuanYan1/PLDG/tree/main. Siyuan Yan, Chi Liu 0002, Lie Ju, Dwarikanath Mahapatra, Brigid Betz-Stablein, Victoria Mar, Monika Janda, H. Peter Soyer, ZongYuan Ge |
IEEE Trans. Medical Imaging | 10 |
| 2025 | ReactFace: Online Multiple Appropriate Facial Reaction Generation in Dyadic InteractionsabstractIn dyadic interaction, predicting the listener's facial reactions is challenging as different reactions could be appropriate in response to the same speaker's behaviour. Previous approaches predominantly treated this task as an interpolation or fitting problem, emphasizing deterministic outcomes but ignoring the diversity and uncertainty of human facial reactions. Furthermore, these methods often failed to model short-range and long-range dependencies within the interaction context, leading to issues in the synchrony and appropriateness of the generated facial reactions. To address these limitations, this paper reformulates the task as an extrapolation or prediction problem, and proposes an novel framework (called ReactFace) to generate multiple different but appropriate facial reactions from a speaker behaviour rather than merely replicating the corresponding listener facial behaviours. Our ReactFace generates multiple different but appropriate photo-realistic human facial reactions by: (i) learning an appropriate facial reaction distribution representing multiple different but appropriate facial reactions; and (ii) synchronizing the generated facial reactions with the speaker verbal and non-verbal behaviours at each time stamp, resulting in realistic 2D facial reaction sequences. Experimental results demonstrate the effectiveness of our approach in generating multiple diverse, synchronized, and appropriate facial reactions from each speaker's behaviour. The quality of the generated facial reactions is intimately tied to the speaker's speech and facial expressions, achieved through our novel speaker-listener interaction modules. Siyang Song, Weicheng Xie 0001, Micol Spitale, ZongYuan Ge, LinLin Shen, Hatice Gunes |
IEEE Trans. Vis. Comput. Graph. | 5 |
| 2024 | Combining Graph Transformers Based Multi-Label Active Learning and Informative Data Augmentation for Chest Xray ClassificationabstractInformative sample selection in active learning (AL) helps a machine learning system attain optimum performance with minimum labeled samples, thus improving human-in-the-loop computer-aided diagnosis systems with limited labeled data. Data augmentation is highly effective for enlarging datasets with less labeled data. Combining informative sample selection and data augmentation should leverage their respective advantages and improve performance of AL systems. We propose a novel approach to combine informative sample selection and data augmentation for multi-label active learning. Conventional informative sample selection approaches have mostly focused on the single-label case which do not perform optimally in the multi-label setting. We improve upon state-of-the-art multi-label active learning techniques by representing disease labels as graph nodes, use graph attention transformers (GAT) to learn more effective inter-label relationships and identify most informative samples. We generate transformations of these informative samples which are also informative. Experiments on public chest xray datasets show improved results over state-of-the-art multi-label AL techniques in terms of classification performance, learning rates, and robustness. We also perform qualitative analysis to determine the realism of generated images. Dwarikanath Mahapatra, Behzad Bozorgtabar, ZongYuan Ge, Mauricio Reyes 0001, Jean-Philippe Thiran |
AAAI | 3 |
| 2024 | Diversified and Personalized Multi-Rater Medical Image SegmentationabstractAnnotation ambiguity due to inherent data uncertainties such as blurred boundaries in medical scans and different observer expertise and preferences has become a major ob-stacle for training deep-learning based medical image segmentation models. To address it, the common practice is to gather multiple annotations from different experts, leading to the setting of multi-rater medical image segmentation. Existing works aim to either merge different annotations into the “groundtruth” that is often unattainable in numerous medical contexts, or generate diverse results, or produce personalized results corresponding to individ-ual expert raters. Here, we bring up a more ambitious goal for multi-rater medical image segmentation, i.e., obtaining both diversified and personalized results. Specifi-cally, we propose a two-stage framework named D-Persona (first Diversification and then Personalization). In Stage I, we exploit multiple given annotations to train a Proba-bilistic U-Net model, with a bound-constrained loss to improve the prediction diversity. In this way, a common latent space is constructed in Stage I, where different latent codes denote diversified expert opinions. Then, in Stage II, we design multiple attention-based projection heads to adaptively query the corresponding expert prompts from the shared latent space, and then perform the personalized medical image segmentation. We evaluated the proposed model on our in-house Nasopharyngeal Carcinoma dataset and the public lung nodule dataset (i.e., LIDC-IDRI). Ex-tensive experiments demonstrated our D-Persona can provide diversified and personalized results at the same time, achieving new SOTA performance for multi-rater medical image segmentation. Our code will be released at https://github.com/ycwu1997/D-Persona. Yicheng Wu 0001, Xiangde Luo, Zhe Xu 0012, Xiaoqing Guo, Lie Ju, ZongYuan Ge, Wenjun Liao, Jianfei Cai 0001 |
CVPR | 6 |
| 2024 | Hunting Attributes: Context Prototype-Aware Learning for Weakly Supervised Semantic SegmentationabstractRecent weakly supervised semantic segmentation (WSSS) methods strive to incorporate contextual knowledge to improve the completeness of class activation maps (CAM). In this work, we argue that the knowledge bias between instances and contexts affects the capability of the prototype to sufficiently understand instance semantics. Inspired by prototype learning theory, we propose leveraging prototype awareness to capture diverse and fine-grained feature attributes of instances. The hypothesis is that contextual prototypes might erroneously activate similar and frequently co-occurring object categories due to this knowledge bias. Therefore, we propose to enhance the prototype representation ability by mitigating the bias to better capture spatial coverage in semantic object regions. With this goal, we present a Context Prototype-Aware Learning (CPAL) strategy, which leverages semantic context to enrich instance comprehension. The core of this method is to accurately capture intra-class variations in object features through context-aware prototypes, facilitating the adaptation to the semantic attributes of various instances. We design feature distribution alignment to optimize prototype awareness, aligning instance feature distributions with dense features. In addition, a unified training framework is proposed to combine label-guided classification supervision and prototypes-guided self-supervision. Experimental results on PASCAL VOC 2012 and MS COCO 2014 show that CPAL significantly improves off-the-shelf methods and achieves state-of-the-art performance. The project is available at https://github.com/Barrett-python/CPAL. Zhongxing Xu, Zhaojun Qu, Wei Feng 0015, Xingjian Jiang, ZongYuan Ge |
CVPR | 6 |
| 2024 | OphNet: A Large-Scale Video Benchmark for Ophthalmic Surgical Workflow Understanding
Peng Xia 0005, Lin Wang 0027, Siyuan Yan, Zhongxing Xu, Yimin Luo, Kaimin Song, Jürgen Leitner, Xuelian Cheng, Chi Liu 0002, Kaijing Zhou, ZongYuan Ge |
ECCV (4) | 14 |
| 2024 | EventRPG: Event Data Augmentation with Relevance Propagation GuidanceabstractEvent camera, a novel bio-inspired vision sensor, has drawn a lot of attention for its low latency, low power consumption, and high dynamic range. Currently, overfitting remains a critical problem in event-based classification tasks for Spiking Neural Network (SNN) due to its relatively weak spatial representation capability. Data augmentation is a simple but efficient method to alleviate overfitting and improve the generalization ability of neural networks, and saliency-based augmentation methods are proven to be effective in the image processing field. However, there is no approach available for extracting saliency maps from SNNs. Therefore, for the first time, we present Spiking Layer-Time-wise Relevance Propagation rule (SLTRP) and Spiking Layer-wise Relevance Propagation rule (SLRP) in order for SNN to generate stable and accurate CAMs and saliency maps. Based on this, we propose EventRPG, which leverages relevance propagation on the spiking neural network for more efficient augmentation. Our proposed method has been evaluated on several SNN structures, achieving state-of-the-art performance in object recognition tasks including N-Caltech101, CIFAR10-DVS, with accuracies of 85.62% and 85.55%, as well as action recognition task SL-Animals with an accuracy of 91.59%. Our code is available at https://github.com/myuansun/EventRPG. Donghao Zhang 0004, ZongYuan Ge, Jia Li 0057, Zheng Fang 0001, Renjing Xu |
ICLR | 3 |
| 2024 | Universal Semi-supervised Learning for Medical Image Classification
Lie Ju, Yicheng Wu 0001, Wei Feng 0015, Lin Wang 0027, Zhuoting Zhu, ZongYuan Ge |
MICCAI (12) | 7 |
| 2024 | TP-DRSeg: Improving Diabetic Retinopathy Lesion Segmentation with Explicit Text-Prompts Assisted SAM
Wenxue Li 0003, Xinyu Xiong, Peng Xia 0005, Lie Ju, ZongYuan Ge |
MICCAI (8) | 5 |
| 2024 | Generalizing to Unseen Domains in Diabetic Retinopathy with Disentangled Representations
Peng Xia 0005, Wenxue Li 0003, Lie Ju, Peibo Duan, Huaxiu Yao, ZongYuan Ge |
MICCAI (10) | 9 |
| 2024 | Fundus2Video: Cross-Modal Angiography Video Generation from Static Fundus Photography with Clinical Knowledge Guidance
Weiyi Zhang 0004, Siyu Huang, Jiancheng Yang, ZongYuan Ge, Yingfeng Zheng, Danli Shi, Mingguang He |
MICCAI (1) | 5 |
| 2024 | CARES: A Comprehensive Benchmark of Trustworthiness in Medical Vision Language ModelsabstractArtificial intelligence has significantly impacted medical applications, particularly with the advent of Medical Large Vision Language Models (Med-LVLMs), sparking optimism for the future of automated and personalized healthcare. However, the trustworthiness of Med-LVLMs remains unverified, posing significant risks for future model deployment. In this paper, we introduce CARES and aim to comprehensively evaluate the Trustworthiness of Med-LVLMs across the medical domain. We assess the trustworthiness of Med-LVLMs across five dimensions, including trustfulness, fairness, safety, privacy, and robustness. CARES comprises about 41K question-answer pairs in both closed and open-ended formats, covering 16 medical image modalities and 27 anatomical regions. Our analysis reveals that the models consistently exhibit concerns regarding trustworthiness, often displaying factual inaccuracies and failing to maintain fairness across different demographic groups. Furthermore, they are vulnerable to attacks and demonstrate a lack of privacy awareness. We publicly release our benchmark and code in https://github.com/richard-peng-xia/CARES. Peng Xia 0005, Juanxi Tian, Yangrui Gong, Ruibo Hou, Zhenbang Wu, Zhiyuan Fan, Yiyang Zhou, Kangyu Zhu, Zhaoyang Wang 0004, Xiao Wang 0044, Xuchao Zhang, Chetan Bansal, Marc Niethammer, Junzhou Huang, Hongtu Zhu, Yun Li 0010, Jimeng Sun 0001, ZongYuan Ge, Gang Li 0001, James Zou 0001, Huaxiu Yao |
NeurIPS | 21 |
| 2024 | GANDALF: Graph-based transformer and Data Augmentation Active Learning Framework with interpretable features for multi-label chest Xray classification
Dwarikanath Mahapatra, Behzad Bozorgtabar, ZongYuan Ge, Mauricio Reyes 0001 |
Medical Image Anal. | 3 |
| 2024 | ALFREDO: Active Learning with FeatuRe disEntangelement and DOmain adaptation for medical image classification
Dwarikanath Mahapatra, Ruwan B. Tennakoon, Yasmeen M. George, Sudipta Roy 0002, Behzad Bozorgtabar, ZongYuan Ge, Mauricio Reyes 0001 |
Medical Image Anal. | 6 |
| 2024 | Hierarchical Knowledge Guided Learning for Real-World Retinal Disease RecognitionabstractIn the real world, medical datasets often exhibit a long-tailed data distribution (i.e., a few classes occupy the majority of the data, while most classes have only a limited number of samples), which results in a challenging long-tailed learning scenario. Some recently published datasets in ophthalmology AI consist of more than 40 kinds of retinal diseases with complex abnormalities and variable morbidity. Nevertheless, more than 30 conditions are rarely seen in global patient cohorts. From a modeling perspective, most deep learning models trained on these datasets may lack the ability to generalize to rare diseases where only a few available samples are presented for training. In addition, there may be more than one disease for the presence of the retina, resulting in a challenging label co-occurrence scenario, also known as multi-label, which can cause problems when some re-sampling strategies are applied during training. To address the above two major challenges, this paper presents a novel method that enables the deep neural network to learn from a long-tailed fundus database for various retinal disease recognition. Firstly, we exploit the prior knowledge in ophthalmology to improve the feature representation using a hierarchy-aware pre-training. Secondly, we adopt an instance-wise class-balanced sampling strategy to address the label co-occurrence issue under the long-tailed medical dataset scenario. Thirdly, we introduce a novel hybrid knowledge distillation to train a less biased representation and classifier. We conducted extensive experiments on four databases, including two public datasets and two in-house databases with more than one million fundus images. The experimental results demonstrate the superiority of our proposed methods with recognition accuracy outperforming the state-of-the-art competitors, especially for these rare diseases. Lie Ju, Lin Wang 0027, Xin Wang 0094, C. Paul Bonnington, ZongYuan Ge |
IEEE Trans. Medical Imaging | 7 |
| 2023 | Unsupervised Domain Adaptation for Medical Image Segmentation by Selective Entropy Constraints and Adaptive Semantic AlignmentabstractGeneralizing a deep learning model to new domains is crucial for computer-aided medical diagnosis systems. Most existing unsupervised domain adaptation methods have made significant progress in reducing the domain distribution gap through adversarial training. However, these methods may still produce overconfident but erroneous results on unseen target images. This paper proposes a new unsupervised domain adaptation framework for cross-modality medical image segmentation. Specifically, We first introduce two data augmentation approaches to generate two sets of semantics-preserving augmented images. Based on the model's predictive consistency on these two sets of augmented images, we identify reliable and unreliable pixels. We then perform a selective entropy constraint: we minimize the entropy of reliable pixels to increase their confidence while maximizing the entropy of unreliable pixels to reduce their confidence. Based on the identified reliable and unreliable pixels, we further propose an adaptive semantic alignment module which performs class-level distribution adaptation by minimizing the distance between same class prototypes between domains, where unreliable pixels are removed to derive more accurate prototypes. We have conducted extensive experiments on the cross-modality cardiac structure segmentation task. The experimental results show that the proposed method significantly outperforms the state-of-the-art comparison algorithms. Our code and data are available at https://github.com/fengweie/SE_ASA. Wei Feng 0015, Lie Ju, Lin Wang 0027, Kaimin Song, ZongYuan Ge |
AAAI | 6 |
| 2023 | Towards Trustable Skin Cancer Diagnosis via Rewriting Model's DecisionabstractDeep neural networks have demonstrated promising performance on image recognition tasks. However, they may heavily rely on confounding factors, using irrelevant artifacts or bias within the dataset as the cue to improve performance. When a model performs decision-making based on these spurious correlations, it can become untrustable and lead to catastrophic outcomes when deployed in the realworld scene. In this paper, we explore and try to solve this problem in the context of skin cancer diagnosis. We introduce a human-in-the-loop framework in the model training process such that users can observe and correct the model's decision logic when confounding behaviors happen. Specifically, our method can automatically discover confounding factors by analyzing the co-occurrence behavior of the samples. It is capable of learning confounding concepts using easily obtained concept exemplars. By mapping the black-box model's feature representation onto an explainable concept space, human users can interpret the concept and intervene via first order-logic instruction. We systematically evaluate our method on our newly crafted, well-controlled skin lesion dataset and several public skin lesion datasets. Experiments show that our method can effectively detect and remove confounding factors from datasets without any prior knowledge about the category distribution and does not require fully annotated concept labels. We also show that our method enables the model to focus on clinical-related concepts, improving the model's performance and trustworthiness during model inference. Siyuan Yan, Dwarikanath Mahapatra, Shekhar Chandra, Monika Janda, H. Peter Soyer, ZongYuan Ge |
CVPR | 8 |
| 2023 | Towards Novel Class Discovery: A Study in Novel Skin Lesions Clustering
Wei Feng 0015, Lie Ju, Lin Wang 0027, Kaimin Song, ZongYuan Ge |
MICCAI (6) | 5 |
| 2023 | Style-Based Manifold for Weakly-Supervised Disease Characteristic Discovery
Siyu Liu 0002, Linfeng Liu 0009, Craig Engstrom, Xuan Vinh To, ZongYuan Ge, Stuart Crozier, Fatima A. Nasrallah, Shekhar Chandra |
MICCAI (5) | 5 |
| 2023 | Class Specific Feature Disentanglement and Text Embeddings for Multi-label Generalized Zero Shot CXR Classification
Dwarikanath Mahapatra, Antonio Jimeno-Yepes, Shiba Kuanar, Sudipta Roy 0002, Behzad Bozorgtabar, Mauricio Reyes 0001, ZongYuan Ge |
MICCAI (2) | 7 |
| 2023 | Privacy-Preserving Early Detection of Epileptic Seizures in Videos
Deval Mehta 0001, Shobi Sivathamboo, Hugh Simpson, Patrick Kwan, Terence J. O'Brien, ZongYuan Ge |
MICCAI (5) | 6 |
| 2023 | EPVT: Environment-Aware Prompt Vision Transformer for Domain Generalization in Skin Lesion Recognition
Siyuan Yan, Chi Liu 0002, Lie Ju, Dwarikanath Mahapatra, Victoria Mar, Monika Janda, H. Peter Soyer, ZongYuan Ge |
MICCAI (7) | 9 |
| 2023 | Retinal Age Estimation with Temporal Fundus Images Enhanced Progressive Label Distribution Learning
Ruiye Chen, Peng Gui, Lie Ju, Xianwen Shang, Zhuoting Zhu, Mingguang He, ZongYuan Ge |
MICCAI (7) | 8 |
| 2023 | EndoSurf: Neural Surface Reconstruction of Deformable Tissues with Stereo Endoscope Videos
Ruyi Zha, Xuelian Cheng, Hongdong Li, Mehrtash Harandi, ZongYuan Ge |
MICCAI (9) | 5 |
| 2023 | NurViD: A Large Expert-Level Video Database for Nursing Procedure Activity UnderstandingabstractThe application of deep learning to nursing procedure activity understanding has the potential to greatly enhance the quality and safety of nurse-patient interactions. By utilizing the technique, we can facilitate training and education, improve quality control, and enable operational compliance monitoring. However, the development of automatic recognition systems in this field is currently hindered by the scarcity of appropriately labeled datasets. The existing video datasets pose several limitations: 1) these datasets are small-scale in size to support comprehensive investigations of nursing activity; 2) they primarily focus on single procedures, lacking expert-level annotations for various nursing procedures and action steps; and 3) they lack temporally localized annotations, which prevents the effective localization of targeted actions within longer video sequences. To mitigate these limitations, we propose NurViD, a large video dataset with expert-level annotation for nursing procedure activity understanding. NurViD consists of over 1.5k videos totaling 144 hours, making it approximately four times longer than the existing largest nursing activity datasets. Notably, it encompasses 51 distinct nursing procedures and 177 action steps, providing a much more comprehensive coverage compared to existing datasets that primarily focus on limited procedures. To evaluate the efficacy of current deep learning methods on nursing activity understanding, we establish three benchmarks on NurViD: procedure recognition on untrimmed videos, procedure and action recognition on trimmed videos, and action detection. Our benchmark and code will be available at https://github.com/minghu0830/NurViD-benchmark. Lin Wang 0027, Siyuan Yan, Don Ma, Qingli Ren, Peng Xia 0005, Wei Feng 0015, Peibo Duan, Lie Ju, ZongYuan Ge |
NeurIPS | 10 |
| 2023 | Medical visual question answering: A survey
Donghao Zhang 0004, Qingyi Tao, Danli Shi, Gholamreza Haffari, Qi Wu 0001, Mingguang He, ZongYuan Ge |
Artif. Intell. Medicine | 8 |
| 2023 | Contrastive pre-training and linear interaction attention-based transformer for universal medical reports generation
Donghao Zhang 0004, Danli Shi, Renjing Xu, Qingyi Tao, Lin Wu 0001, Mingguang He, ZongYuan Ge |
J. Biomed. Informatics | 8 |
| 2023 | Dual-branch cross-dimensional self-attention-based imputation model for multivariate time seriesabstractIn real-world scenarios, partial information losses of multivariate time series degrade the time series analysis. Hence, the time series imputation technique has been adopted to compensate for the missing values. Existing methods focus on investigating temporal correlations, cross-variable correlations, and bidirectional dynamics of time series, and most of these methods rely on recurrent neural networks (RNNs) to capture temporal dependency. However, the RNN-based models suffer from the common problems of slow speed and high complexity when dealing with long-term dependency. While some self-attention-based models without any recurrent structures can tackle long-term dependency with parallel computing, they do not fully learn and utilize correlations across the temporal and cross-variable dimensions. To address the limitations of existing methods, we propose a novel so-called dual-branch cross-dimensional self-attention-based imputation (DCSAI) model for multivariate time series, which is capable of performing global and auxiliary cross-dimensional analyses when imputing the missing values. In particular, this model contains masked multi-head self-attention-based encoders aligned with auxiliary generators to obtain global and auxiliary correlations in two dimensions, and these correlations are then combined into one final representation through three weighted combinations. Extensive experiments are presented to show that our model performs better than other state-of-the-art benchmarkers on three real-world public datasets under various missing rates. Furthermore, ablation study results demonstrate the efficacy of each component of the model. Le Fang 0001, Wei Xiang 0001, Yuan Zhou 0006, Juan Fang 0004, Lianhua Chi, ZongYuan Ge |
Knowl. Based Syst. | 6 |
| 2023 | Normal vibration distribution search-based differential evolution algorithm for multimodal biomedical image registrationabstractIn linear registration, a floating image is spatially aligned with a reference image after performing a series of linear metric transformations. Additionally, linear registration is mainly considered a preprocessing version of nonrigid registration. To better accomplish the task of finding the optimal transformation in pairwise intensity-based medical image registration, in this work, we present an optimization algorithm called the normal vibration distribution search-based differential evolution algorithm (NVSA), which is modified from the Bernstein search-based differential evolution (BSD) algorithm. We redesign the search pattern of the BSD algorithm and import several control parameters as part of the fine-tuning process to reduce the difficulty of the algorithm. In this study, 23 classic optimization functions and 16 real-world patients (resulting in 41 multimodal registration scenarios) are used in experiments performed to statistically investigate the problem solving ability of the NVSA. Nine metaheuristic algorithms are used in the conducted experiments. When compared to the commonly utilized registration methods, such as ANTS, Elastix, and FSL, our method achieves better registration performance on the RIRE dataset. Moreover, we prove that our method can perform well with or without its initial spatial transformation in terms of different evaluation indicators, demonstrating its versatility and robustness for various clinical needs and applications. This study establishes the idea that metaheuristic-based methods can better accomplish linear registration tasks than the frequently used approaches; the proposed method demonstrates promise that it can solve real-world clinical and service problems encountered during nonrigid registration as a preprocessing approach.The source code of the NVSA is publicly available at https://github.com/PengGui-N/NVSA. Peng Gui, Fazhi He, Bingo Wing-Kuen Ling, Dengyi Zhang, ZongYuan Ge |
Neural Comput. Appl. | 5 |
| 2023 | Generative Metric Learning for Adversarially Robust Open-world Person Re-IdentificationabstractThe vulnerability of re-identification (re-ID) models under adversarial attacks is of significant concern as criminals may use adversarial perturbations to evade surveillance systems. Unlike a closed-world re-ID setting (i.e., a fixed number of training categories), a reliable re-ID system in the open world raises the concern of training a robust yet discriminative classifier, which still shows robustness in the context of unknown examples of an identity. In this work, we improve the robustness of open-world re-ID models by proposing a generative metric learning approach to generate adversarial examples that are regularized to produce robust distance metric. The proposed approach leverages the expressive capability of generative adversarial networks to defend the re-ID models against feature disturbance attacks. By generating the target people variants and sampling the triplet units for metric learning, our learned distance metrics are regulated to produce accurate predictions in the feature metric space. Experimental results on the three re-ID datasets, i.e., Market-1501, DukeMTMC-reID, and MSMT17 demonstrate the robustness of our method. Deyin Liu, Lin Wu 0001, Richang Hong, ZongYuan Ge, Jialie Shen 0001, Farid Boussaïd, Mohammed Bennamoun |
ACM Trans. Multim. Comput. Commun. Appl. | 4 |
| 2022 | Learning Network Architecture for Open-Set RecognitionabstractGiven the incomplete knowledge of classes that exist in the world, Open-set Recognition (OSR) enables networks to identify and reject the unseen classes after training. This problem of breaking the common closed-set assumption is far from being solved. Recent studies focus on designing new losses, neural network encoding structures, and calibration methods to optimize a feature space for OSR relevant tasks. In this work, we make the first attempt to tackle OSR by searching the architecture of a Neural Network (NN) under the open-set assumption. In contrast to the prior arts, we develop a mechanism to both search the architecture of the network and train a network suitable for tackling OSR. Inspired by the compact abating probability (CAP) model, which is theoretically proven to reduce the open space risk, we regularize the searching space by VAE contrastive learning. To discover a more robust structure for OSR, we propose Pseudo Auxiliary Searching (PAS), in which we split a pretended set of know-unknown classes from the original training set in the searching phase, hence enabling the super-net to explore an effective architecture that can handle unseen classes in advance. We demonstrate the benefits of this learning pipeline on 5 OSR datasets, including MNIST, SVHN, CIFAR10, CIFARAdd10, and CIFARAdd50, where our approach outperforms prior state-of-the-art networks designed by humans. To spark research in this field, our code is available at https://github.com/zxl101/NAS OSR. Xuelian Cheng, Donghao Zhang 0004, C. Paul Bonnington, ZongYuan Ge |
AAAI | 5 |
| 2022 | Implicit Motion Handling for Video Camouflaged Object DetectionabstractWe propose a new video camouflaged object detection (VCOD) framework that can exploit both short-term dynamics and long-term temporal consistency to detect camouflaged objects from video frames. An essential property of camouflaged objects is that they usually exhibit patterns similar to the background and thus make them hard to identify from still images. Therefore, effectively handling temporal dynamics in videos becomes the key for the VCOD task as the camouflaged objects will be noticeable when they move. However, current VCOD methods often leverage homography or optical flows to represent motions, where the detection error may accumulate from both the motion estimation error and the segmentation error. On the other hand, our method unifies motion estimation and object segmentation within a single optimization framework. Specifically, we build a dense correlation volume to implicitly capture motions between neighbouring frames and utilize the final segmentation supervision to optimize the implicit motion estimation and segmentation jointly. Furthermore, to enforce temporal consistency within a video sequence, we jointly utilize a spatio-temporal transformer to refine the short-term predictions. Extensive experiments on VCOD benchmarks demonstrate the architectural effectiveness of our approach. We also provide a large-scale VCOD dataset named MoCA-Mask with pixel-level handcrafted ground-truth masks and construct a comprehensive VCOD bench-mark with previous methods to facilitate research in this direction. Dataset Link: https://xueliancheng.github.io/SLT-Net-project. Xuelian Cheng, Huan Xiong, Deng-Ping Fan, Yiran Zhong, Mehrtash Harandi, Tom Drummond, ZongYuan Ge |
CVPR | 7 |
| 2022 | Node Representation Learning in Graph via Node-to-Neighbourhood Mutual Information MaximizationabstractThe key towards learning informative node representations in graphs lies in how to gain contextual information from the neighbourhood. In this work, we present a simple-yet-effective self-supervised node representation learning strategy via directly maximizing the mutual information between the hidden representations of nodes and their neighbourhood, which can be theoretically justified by its link to graph smoothing. Following InfoNCE, our framework is optimized via a surrogate contrastive loss, where the positive selection underpins the quality and efficiency of rep-resentation learning. To this end, we propose a topology-aware positive sampling strategy, which samples positives from the neighbourhood by considering the structural dependencies between nodes and thus enables positive selection upfront. In the extreme case when only one positive is sampled, we fully avoid expensive neighbourhood aggregation. Our methods achieve promising performance on various node classification datasets. It is also worth mentioning by applying our loss function to MLP based node encoders, our methods can be orders of faster than existing solutions. Our codes and supplementary materials are available at https://github.com/dongwei156/n2n. Wei Dong 0010, Junsheng Wu, ZongYuan Ge, Peng Wang 0023 |
CVPR | 4 |
| 2022 | Deep Laparoscopic Stereo Matching with Transformers
Xuelian Cheng, Yiran Zhong, Mehrtash Harandi, Tom Drummond, Zhiyong Wang 0001, ZongYuan Ge |
MICCAI (8) | 6 |
| 2022 | Unsupervised Domain Adaptive Fundus Image Segmentation with Category-Level Regularization
Wei Feng 0015, Lin Wang 0027, Lie Ju, Xin Wang 0094, ZongYuan Ge |
MICCAI (2) | 7 |
| 2022 | Flexible Sampling for Long-Tailed Skin Lesion Classification
Lie Ju, Yicheng Wu 0001, Lin Wang 0027, Xin Wang 0094, C. Paul Bonnington, ZongYuan Ge |
MICCAI (3) | 8 |
| 2022 | Camera Adaptation for Fundus-Image-Based CVD Risk Estimation
Danli Shi, Donghao Zhang 0004, Xianwen Shang, Mingguang He, ZongYuan Ge |
MICCAI (2) | 6 |
| 2022 | Out-of-Distribution Detection for Long-Tailed and Fine-Grained Skin Lesion Images
Deval Mehta 0001, Yaniv Gal, Adrian Bowling, C. Paul Bonnington, ZongYuan Ge |
MICCAI (1) | 5 |
| 2022 | Exploring Smoothness and Class-Separation for Semi-supervised Medical Image Segmentation
Yicheng Wu 0001, Qianyi Wu, ZongYuan Ge, Jianfei Cai 0001 |
MICCAI (5) | 4 |
| 2022 | Skin Lesion Recognition with Class-Hierarchy Regularized Hyperbolic Embeddings
Toàn D. Nguyên, Yaniv Gal, Lie Ju, Shekhar Chandra, Lei Zhang 0095, C. Paul Bonnington, Victoria Mar, Zhiyong Wang 0001, ZongYuan Ge |
MICCAI (3) | 10 |
| 2022 | RBP-TSTL is a two-stage transfer learning framework for genome-scale prediction of RNA-binding proteinsabstractRNA binding proteins (RBPs) are critical for the post-transcriptional control of RNAs and play vital roles in a myriad of biological processes, such as RNA localization and gene regulation. Therefore, computational methods that are capable of accurately identifying RBPs are highly desirable and have important implications for biomedical and biotechnological applications. Here, we propose a two-stage deep transfer learning-based framework, termed RBP-TSTL, for accurate prediction of RBPs. In the first stage, the knowledge from the self-supervised pre-trained model was extracted as feature embeddings and used to represent the protein sequences, while in the second stage, a customized deep learning model was initialized based on an annotated pre-training RBPs dataset before being fine-tuned on each corresponding target species dataset. This two-stage transfer learning framework can enable the RBP-TSTL model to be effectively trained to learn and improve the prediction performance. Extensive performance benchmarking of the RBP-TSTL models trained using the features generated by the self-supervised pre-trained model and other models trained using hand-crafting encoding features demonstrated the effectiveness of the proposed two-stage knowledge transfer strategy based on the self-supervised pre-trained models. Using the best-performing RBP-TSTL models, we further conducted genome-scale RBP predictions for Homo sapiens, Arabidopsis thaliana, Escherichia coli, and Salmonella and established a computational compendium containing all the predicted putative RBPs candidates. We anticipate that the proposed RBP-TSTL approach will be explored as a useful tool for the characterization of RNA-binding proteins and exploration of their sequence-structure-function relationships. Xinxin Peng, Xiaoyu Wang 0016, Yuming Guo 0001, ZongYuan Ge, Fuyi Li, Xin Gao 0001, Jiangning Song |
Briefings Bioinform. | 4 |
| 2022 | ASPIRER: a new computational approach for identifying non-classical secreted proteins based on deep learningabstractProtein secretion has a pivotal role in many biological processes and is particularly important for intercellular communication, from the cytoplasm to the host or external environment. Gram-positive bacteria can secrete proteins through multiple secretion pathways. The non-classical secretion pathway has recently received increasing attention among these secretion pathways, but its exact mechanism remains unclear. Non-classical secreted proteins (NCSPs) are a class of secreted proteins lacking signal peptides and motifs. Several NCSP predictors have been proposed to identify NCSPs and most of them employed the whole amino acid sequence of NCSPs to construct the model. However, the sequence length of different proteins varies greatly. In addition, not all regions of the protein are equally important and some local regions are not relevant to the secretion. The functional regions of the protein, particularly in the N- and C-terminal regions, contain important determinants for secretion. In this study, we propose a new hybrid deep learning-based framework, referred to as ASPIRER, which improves the prediction of NCSPs from amino acid sequences. More specifically, it combines a whole sequence-based XGBoost model and an N-terminal sequence-based convolutional neural network model; 5-fold cross-validation and independent tests demonstrate that ASPIRER achieves superior performance than existing state-of-the-art approaches. The source code and curated datasets of ASPIRER are publicly available at https://github.com/yanwu20/ASPIRER/. ASPIRER is anticipated to be a useful tool for improved prediction of novel putative NCSPs from sequences information and prioritization of candidate proteins for follow-up experimental validation. Xiaoyu Wang 0016, Fuyi Li, Jing Xu 0008, Jia Rong, Geoffrey I. Webb, ZongYuan Ge, Jian Li 0052, Jiangning Song |
Briefings Bioinform. | 6 |
| 2022 | Mutual consistency learning for semi-supervised medical image segmentation
Yicheng Wu 0001, ZongYuan Ge, Donghao Zhang 0004, Minfeng Xu, Lei Zhang 0006, Yong Xia 0001, Jianfei Cai 0001 |
Medical Image Anal. | 2 |
| 2022 | ZeroNAS: Differentiable Generative Adversarial Networks Search for Zero-Shot LearningabstractIn recent years, remarkable progress in zero-shot learning (ZSL) has been achieved by generative adversarial networks (GAN). To compensate for the lack of training samples in ZSL, a surge of GAN architectures have been developed by human experts through trial-and-error testing. Despite their efficacy, however, there is still no guarantee that these hand-crafted models can consistently achieve good performance across diversified datasets or scenarios. Accordingly, in this paper, we turn to neural architecture search (NAS) and make the first attempt to bring NAS techniques into the ZSL realm. Specifically, we propose a differentiable GAN architecture search method over a specifically designed search space for zero-shot learning, referred to as ZeroNAS. Considering the relevance and balance of the generator and discriminator, ZeroNAS jointly searches their architectures in a min-max player game via adversarial training. Extensive experiments conducted on four widely used benchmark datasets demonstrate that ZeroNAS is capable of discovering desirable architectures that perform favorably against state-of-the-art ZSL and generalized zero-shot learning (GZSL) approaches. Source code is at https://github.com/caixiay/ZeroNAS. Caixia Yan, Xiaojun Chang, Zhihui Li 0001, Weili Guan, ZongYuan Ge, Lei Zhu 0002 |
IEEE Trans. Pattern Anal. Mach. Intell. | 5 |
| 2022 | Contextual ensemble network for semantic segmentation
Quan Zhou 0004, Xiaofu Wu, Suofei Zhang, Bin Kang, ZongYuan Ge, Longin Jan Latecki |
Pattern Recognit. | 5 |
| 2022 | Auto-FSL: Searching the Attribute Consistent Network for Few-Shot LearningabstractPrevailing deep methods for image recognition require massive labeled samples in each visual category for training. However, large amounts of data annotations are timeconsuming, and some uncommon categories only have rare samples available. For this issue, we focus on more challenging few-shot learning (FSL) task, where just few labeled images are used in the training stage. Existing FSL models are constructed with various convolutional neural networks (CNNs), which are trained on an auxiliary base dataset and evaluated for new fewshot predictions on a novel dataset. The performance of these models is difficult to break through because of the domain shift between base and novel datasets and the monotonous network architectures. Considering that, we propose a novel automatic attribute consistent network called Auto-ACNet to overcome the above problems. On one hand, Auto-ACNet utilizes the attribute information about base and novel categories to guide the procedure of representation learning. It introduces the consistent and non-consistent subnets to capture the common and different attributes of image pair, which helps to mitigate the domain shift problem. On the other hand, the architecture of Auto-ACNet is searched with the popular neural architecture search (NAS) technique DARTS, for obtaining a superior FSL network automatically. And the DARTS’s search space is improved by adding the position-aware module to extract the attribute characteristics better. Extensive experimental results on two datasets indicate that the proposed Auto-ACNet achieves significant improvement over the state-of-the-art competitors in this literature. Lingling Zhang 0005, Xiaojun Chang, Jun Liu 0002, ZongYuan Ge |
IEEE Trans. Circuits Syst. Video Technol. | 5 |
| 2022 | Pseudo-Pair Based Self-Similarity Learning for Unsupervised Person Re-IdentificationabstractPerson re-identification (re-ID) is of great importance to video surveillance systems by estimating the similarity between a pair of cross-camera person shorts. Current methods for estimating such similarity require a large number of labeled samples for supervised training. In this paper, we present a pseudo-pair based self-similarity learning approach for unsupervised person re-ID without human annotations. Unlike conventional unsupervised re-ID methods that use pseudo labels based on global clustering, we construct patch surrogate classes as initial supervision, and propose to assign pseudo labels to images through the pairwise gradient-guided similarity separation. This can cluster images in pseudo pairs, and the pseudos can be updated during training. Based on pseudo pairs, we propose to improve the generalization of similarity function via a novel self-similarity learning:it learns local discriminative features from individual images via intra-similarity, and discovers the patch correspondence across images via inter-similarity. The intra-similarity learning is based on channel attention to detect diverse local features from an image. The inter-similarity learning employs a deformable convolution with a non-local block to align patches for cross-image similarity. Experimental results on several re-ID benchmark datasets demonstrate the superiority of the proposed method over the state-of-the-arts. Lin Wu 0001, Deyin Liu, Dapeng Chen, ZongYuan Ge, Farid Boussaïd, Mohammed Bennamoun, Jialie Shen 0001 |
IEEE Trans. Image Process. | 5 |
| 2022 | Improving Medical Images Classification With Label Noise Using Dual-Uncertainty EstimationabstractDeep neural networks are known to be data-driven and label noise can have a marked impact on model performance. Recent studies have shown great robustness to classic image recognition even under a high noisy rate. In medical applications, learning from datasets with label noise is more challenging since medical imaging datasets tend to have instance-dependent noise (IDN) and suffer from high observer variability. In this paper, we systematically discuss the two common types of label noise in medical images - disagreement label noise from inconsistency expert opinions and single-target label noise from biased aggregation of individual annotations. We then propose an uncertainty estimation-based framework to handle these two label noise amid the medical image classification task. We design a dual-uncertainty estimation approach to measure the disagreement label noise and single-target label noise via improved Direct Uncertainty Prediction and Monte-Carlo-Dropout. A boosting-based curriculum training procedure is later introduced for robust learning. We demonstrate the effectiveness of our method by conducting extensive experiments on three different diseases with synthesized and real-world label noise: skin lesions, prostate cancer, and retinal diseases. We also release a large re-engineered database that consists of annotations from more than ten ophthalmologists with an unbiased golden standard dataset for evaluation and benchmarking. The dataset is available at https://mmai.group/peoples/julie/. Lie Ju, Xin Wang 0094, Lin Wang 0027, Dwarikanath Mahapatra, Quan Zhou 0004, Tongliang Liu, ZongYuan Ge |
IEEE Trans. Medical Imaging | 8 |
| 2022 | Self-Supervised Generalized Zero Shot Learning for Medical Image Classification Using Novel Interpretable Saliency MapsabstractIn many real world medical image classification settings, access to samples of all disease classes is not feasible, affecting the robustness of a system expected to have high performance in analyzing novel test data. This is a case of generalized zero shot learning (GZSL) aiming to recognize seen and unseen classes. We propose a GZSL method that uses self supervised learning (SSL) for: 1) selecting representative vectors of disease classes; and 2) synthesizing features of unseen classes. We also propose a novel approach to generate GradCAM saliency maps that highlight diseased regions with greater accuracy. We exploit information from the novel saliency maps to improve the clustering process by: 1) Enforcing the saliency maps of different classes to be different; and 2) Ensuring that clusters in the space of image and saliency features should yield class centroids having similar semantic information. This ensures the anchor vectors are representative of each class. Different from previous approaches, our proposed approach does not require class attribute vectors which are essential part of GZSL methods for natural images but are not available for medical images. Using a simple architecture the proposed method outperforms state of the art SSL based GZSL performance for natural images as well as multiple types of medical images. We also conduct many ablation studies to investigate the influence of different loss terms in our method. Dwarikanath Mahapatra, ZongYuan Ge, Mauricio Reyes 0001 |
IEEE Trans. Medical Imaging | 2 |
| 2022 | Early Melanoma Diagnosis With Sequential Dermoscopic ImagesabstractDermatologists often diagnose or rule out early melanoma by evaluating the follow-up dermoscopic images of skin lesions. However, existing algorithms for early melanoma diagnosis are developed using single time-point images of lesions. Ignoring the temporal, morphological changes of lesions can lead to misdiagnosis in borderline cases. In this study, we propose a framework for automated early melanoma diagnosis using sequential dermoscopic images. To this end, we construct our method in three steps. First, we align sequential dermoscopic images of skin lesions using estimated Euclidean transformations, extract the lesion growth region by computing image differences among the consecutive images, and then propose a spatio-temporal network to capture the dermoscopic changes from aligned lesion images and the corresponding difference images. Finally, we develop an early diagnosis module to compute probability scores of malignancy for lesion images over time. We collected 179 serial dermoscopic imaging data from 122 patients to verify our method. Extensive experiments show that the proposed model outperforms other commonly used sequence models. We also compared the diagnostic results of our model with those of seven experienced dermatologists and five registrars. Our model achieved higher diagnostic accuracy than clinicians (63.69% vs. 54.33%, respectively) and provided an earlier diagnosis of melanoma (60.7% vs. 32.7% of melanoma correctly diagnosed on the first follow-up images). These results demonstrate that our model can be used to identify melanocytic lesions that are at high-risk of malignant transformation earlier in the disease process and thereby redefine what is possible in the early detection of melanoma. Jennifer Nguyen, Toàn D. Nguyên, John Kelly, Catriona A. McLean, C. Paul Bonnington, Lei Zhang 0095, Victoria Mar, ZongYuan Ge |
IEEE Trans. Medical Imaging | 9 |
| 2021 | Vision-Language Navigation with Random Environmental MixupabstractVision-language Navigation (VLN) tasks require an agent to navigate step-by-step while perceiving the visual observations and comprehending a natural language instruction. Large data bias, which is caused by the disparity ratio between the small data scale and large navigation space, makes the VLN task challenging. Previous works have proposed various data augmentation methods to reduce data bias. However, these works do not explicitly reduce the data bias across different house scenes. Therefore, the agent would overfit to the seen scenes and achieve poor navigation performance in the unseen scenes. To tackle this problem, we propose the Random Environmental Mixup (REM) method, which generates cross-connected house scenes as augmented data via mixuping environment. Specifically, we first select key viewpoints according to the room connection graph for each scene. Then, we cross-connect the key views of different scenes to construct augmented scenes. Finally, we generate augmented instruction-path pairs in the cross-connected scenes. The experimental results on benchmark datasets demonstrate that our augmentation data via REM help the agent reduce its performance gap between the seen and unseen environment and improve the overall performance, making our model the best existing approach on the standard VLN benchmark. Chong Liu 0002, Fengda Zhu, Xiaojun Chang, Xiaodan Liang, ZongYuan Ge, Yidong Shen |
ICCV | 5 |
| 2021 | Robust early-learning: Hindering the memorization of noisy labels
Xiaobo Xia, Tongliang Liu, Bo Han 0003, Chen Gong 0002, Nannan Wang 0001, ZongYuan Ge |
ICLR | 6 |
| 2021 | Retrospective Class Incremental LearningabstractExisting works study the Class Incremental learning (CIL) problem with the assumption that the data for previous classes are absent, or only a small subset of samples (known as exemplars) are accessible. Differently, we propose a new and practical setting called retrospective CIL, where all the previous data are accessible, but with bounded training budgets for old data replay. Since only a small subset of old samples can be replayed, it brings a new research problem, i.e., dynamically sampling old data along the incremental training process. As incremental learning particularly suffers from catastrophic forgetting, we propose to use the forgettability of the old samples as the sampling priorities to favour the forgotten samples during the dynamic sampling process. To achieve this, we introduce a forgetting rate metric with graph- based propagation to estimate the sample forgettability. The proposed method brings improvements on two benchmark datasets. Qingyi Tao, Chen Change Loy, Jianfei Cai 0001, ZongYuan Ge, Simon See |
ICME | 4 |
| 2021 | Relational Subsets Knowledge Distillation for Long-Tailed Retinal Diseases Recognition
Lie Ju, Xin Wang 0094, Lin Wang 0027, Tongliang Liu, Tom Drummond, Dwarikanath Mahapatra, ZongYuan Ge |
MICCAI (8) | 8 |
| 2021 | Medical Matting: A New Perspective on Medical Segmentation with Uncertainty
Lin Wang 0027, Lie Ju, Donghao Zhang 0004, Xin Wang 0094, Wanji He, Yelin Huang, Xiufen Ye, ZongYuan Ge |
MICCAI (3) | 11 |
| 2021 | Semi-supervised Left Atrium Segmentation with Mutual Consistency Training
Yicheng Wu 0001, Minfeng Xu, ZongYuan Ge, Jianfei Cai 0001, Lei Zhang 0006 |
MICCAI (2) | 3 |
| 2021 | End-to-End Ugly Duckling Sign Detection for Melanoma Identification with Transformers
Victoria Mar, Anders Eriksson, Shekhar Chandra, C. Paul Bonnington, Lei Zhang 0095, ZongYuan Ge |
MICCAI (7) | 7 |
| 2021 | Computational prediction and interpretation of both general and specific types of promoters in Escherichia coli by exploiting a stacked ensemble-learning frameworkabstractPromoters are short consensus sequences of DNA, which are responsible for transcription activation or the repression of all genes. There are many types of promoters in bacteria with important roles in initiating gene transcription. Therefore, solving promoter-identification problems has important implications for improving the understanding of their functions. To this end, computational methods targeting promoter classification have been established; however, their performance remains unsatisfactory. In this study, we present a novel stacked-ensemble approach (termed SELECTOR) for identifying both promoters and their respective classification. SELECTOR combined the composition of k-spaced nucleic acid pairs, parallel correlation pseudo-dinucleotide composition, position-specific trinucleotide propensity based on single-strand, and DNA strand features and using five popular tree-based ensemble learning algorithms to build a stacked model. Both 5-fold cross-validation tests using benchmark datasets and independent tests using the newly collected independent test dataset showed that SELECTOR outperformed state-of-the-art methods in both general and specific types of promoter prediction in Escherichia coli. Furthermore, this novel framework provides essential interpretations that aid understanding of model success by leveraging the powerful Shapley Additive exPlanation algorithm, thereby highlighting the most important features relevant for predicting both general and specific types of promoters and overcoming the limitations of existing 'Black-box' approaches that are unable to reveal causal relationships from large amounts of initially encoded features. Fuyi Li, ZongYuan Ge, Yanwei Yue, Morihiro Hayashida, Abdelkader Baggag, Halima Bensmail, Jiangning Song |
Briefings Bioinform. | 3 |
| 2021 | OCTID: a one-class learning-based Python package for tumor image detectionabstractMOTIVATION: Tumor tile selection is a necessary prerequisite in patch-based cancer whole slide image analysis, which is labor-intensive and requires expertise. Whole slides are annotated as tumor or tumor free, but tiles within a tumor slide are not. As all tiles within a tumor free slide are tumor free, these can be used to capture tumor-free patterns using the one-class learning strategy. RESULTS: We present a Python package, termed OCTID, which combines a pretrained convolutional neural network (CNN) model, Uniform Manifold Approximation and Projection (UMAP) and one-class support vector machine to achieve accurate tumor tile classification using a training set of tumor free tiles. Benchmarking experiments on four H&E image datasets achieved remarkable performance in terms of F1-score (0.90 ± 0.06), Matthews correlation coefficient (0.93 ± 0.05) and accuracy (0.94 ± 0.03). AVAILABILITY AND IMPLEMENTATION: Detailed information can be found in the Supplementary File. SUPPLEMENTARY INFORMATION: Supplementary data are available at Bioinformatics online. Yanan Wang 0003, Litao Yang, Geoffrey I. Webb, ZongYuan Ge, Jiangning Song |
Bioinform. | 4 |
| 2021 | One-Shot Neural Architecture Search: Maximising Diversity to Overcome Catastrophic ForgettingabstractOne-shot neural architecture search (NAS) has recently become mainstream in the NAS community because it significantly improves computational efficiency through weight sharing. However, the supernet training paradigm in one-shot NAS introduces catastrophic forgetting, where each step of the training can deteriorate the performance of other architectures that contain partially-shared weights with current architecture. To overcome this problem of catastrophic forgetting, we formulate supernet training for one-shot NAS as a constrained continual learning optimization problem such that learning the current architecture does not degrade the validation accuracy of previous architectures. The key to solving this constrained optimization problem is a novelty search based architecture selection (NSAS) loss function that regularizes the supernet training by using a greedy novelty search method to find the most representative subset. We applied the NSAS loss function to two one-shot NAS baselines and extensively tested them on both a common search space and a NAS benchmark dataset. We further derive three variants based on the NSAS loss function, the NSAS with depth constrain (NSAS-C) to improve the transferability, and NSAS-G and NSAS-LG to handle the situation with a limited number of constraints. The experiments on the common NAS search space demonstrate that NSAS and it variants improve the predictive ability of supernet training in one-shot NAS with remarkable and efficient performance on the CIFAR-10, CIFAR-100, and ImageNet datasets. The results with the NAS benchmark dataset also confirm the significant improvements these one-shot NAS baselines can make. Miao Zhang 0022, Huiqi Li, Shirui Pan, Xiaojun Chang, Chuan Zhou 0001, ZongYuan Ge, Steven W. Su |
IEEE Trans. Pattern Anal. Mach. Intell. | 6 |
| 2021 | Synergic Adversarial Label Learning for Grading Retinal Diseases via Knowledge Distillation and Multi-Task LearningabstractThe need for comprehensive and automated screening methods for retinal image classification has long been recognized. Well-qualified doctors annotated images are very expensive and only a limited amount of data is available for various retinal diseases such as diabetic retinopathy (DR) and age-related macular degeneration (AMD). Some studies show that some retinal diseases such as DR and AMD share some common features like haemorrhages and exudation but most classification algorithms only train those disease models independently when the only single label for one image is available. Inspired by multi-task learning where additional monitoring signals from various sources is beneficial to train a robust model. We propose a method called synergic adversarial label learning (SALL) which leverages relevant retinal disease labels in both semantic and feature space as additional signals and train the model in a collaborative manner using knowledge distillation. Our experiments on DR and AMD fundus image classification task demonstrate that the proposed method can significantly improve the accuracy of the model for grading diseases by 5.91% and 3.69% respectively. In addition, we conduct additional experiments to show the effectiveness of SALL from the aspects of reliability and interpretability in the context of medical imaging application. Lie Ju, Xin Wang 0094, Huimin Lu 0001, Dwarikanath Mahapatra, C. Paul Bonnington, ZongYuan Ge |
IEEE J. Biomed. Health Informatics | 7 |
| 2021 | Keyframe Extraction From Laparoscopic Videos via Diverse and Weighted Dictionary SelectionabstractLaparoscopic videos have been increasingly acquired for various purposes including surgical training and quality assurance, due to the wide adoption of laparoscopy in minimally invasive surgeries. However, it is very time consuming to view a large amount of laparoscopic videos, which prevents the values of laparoscopic video archives from being well exploited. In this paper, a dictionary selection based video summarization method is proposed to effectively extract keyframes for fast access of laparoscopic videos. Firstly, unlike the low-level feature used in most existing summarization methods, deep features are extracted from a convolutional neural network to effectively represent video frames. Secondly, based on such a deep representation, laparoscopic video summarization is formulated as a diverse and weighted dictionary selection model, in which image quality is taken into account to select high quality keyframes, and a diversity regularization term is added to reduce redundancy among the selected keyframes. Finally, an iterative algorithm with a rapid convergence rate is designed for model optimization, and the convergence of the proposed method is also analyzed. Experimental results on a recently released laparoscopic dataset demonstrate the clear superiority of the proposed methods. The proposed method can facilitate the access of key information in surgeries, training of junior clinicians, explanations to patients, and archive of case files. Mingyang Ma 0004, Shaohui Mei, Shuai Wan, Zhiyong Wang 0001, ZongYuan Ge, Vincent Lam, David Dagan Feng |
IEEE J. Biomed. Health Informatics | 5 |
| 2021 | Leveraging Regular Fundus Images for Training UWF Fundus Diagnosis Models via Adversarial Learning and Pseudo-LabelingabstractRecently, ultra-widefield (UWF) 200° fundus imaging by Optos cameras has gradually been introduced because of its broader insights for detecting more information on the fundus than regular 30° - 60° fundus cameras. Compared with UWF fundus images, regular fundus images contain a large amount of high-quality and well-annotated data. Due to the domain gap, models trained by regular fundus images to recognize UWF fundus images perform poorly. Hence, given that annotating medical data is labor intensive and time consuming, in this paper, we explore how to leverage regular fundus images to improve the limited UWF fundus data and annotations for more efficient training. We propose the use of a modified cycle generative adversarial network (CycleGAN) model to bridge the gap between regular and UWF fundus and generate additional UWF fundus images for training. A consistency regularization term is proposed in the loss of the GAN to improve and regulate the quality of the generated data. Our method does not require that images from the two domains be paired or even that the semantic labels be the same, which provides great convenience for data collection. Furthermore, we show that our method is robust to noise and errors introduced by the generated unlabeled data with the pseudo-labeling technique. We evaluated the effectiveness of our methods on several common fundus diseases and tasks, such as diabetic retinopathy (DR) classification, lesion detection and tessellated fundus segmentation. The experimental results demonstrate that our proposed method simultaneously achieves superior generalizability of the learned representations and performance improvements in multiple tasks. Lie Ju, Xin Wang 0094, C. Paul Bonnington, Tom Drummond, ZongYuan Ge |
IEEE Trans. Medical Imaging | 6 |
| 2020 | Geometry-Constrained Car Recognition Using a 3D Perspective NetworkabstractWe present a novel learning framework for vehicle recognition from a single RGB image. Unlike existing methods which only use attention mechanisms to locate 2D discriminative information, our work learns a novel 3D perspective feature representation of a vehicle, which is then fused with 2D appearance feature to predict the category. The framework is composed of a global network (GN), a 3D perspective network (3DPN), and a fusion network. The GN is used to locate the region of interest (RoI) and generate the 2D global feature. With the assistance of the RoI, the 3DPN estimates the 3D bounding box under the guidance of the proposed vanishing point loss, which provides a perspective geometry constraint. Then the proposed 3D representation is generated by eliminating the viewpoint variance of the 3D bounding box using perspective transformation. Finally, the 3D and 2D feature are fused to predict the category of the vehicle. We present qualitative and quantitative results on the vehicle classification and verification tasks in the BoxCars dataset. The results demonstrate that, by learning such a concise 3D representation, we can achieve superior performance to methods that only use 2D information while retain 3D meaningful information without the challenge of requiring a 3D CAD model. ZongYuan Ge, Simon Denman, Sridha Sridharan, Clinton Fookes |
AAAI | 2 |
| 2020 | ZSTAD: Zero-Shot Temporal Activity DetectionabstractAn integral part of video analysis and surveillance is temporal activity detection, which means to simultaneously recognize and localize activities in long untrimmed videos. Currently, the most effective methods of temporal activity detection are based on deep learning, and they typically perform very well with large scale annotated videos for training. However, these methods are limited in real applications due to the unavailable videos about certain activity classes and the time-consuming data annotation. To solve this challenging problem, we propose a novel task setting called zero-shot temporal activity detection (ZSTAD), where activities that have never been seen in training can still be detected. We design an end-to-end deep network based on R-C3D as the architecture for this solution. The proposed network is optimized with an innovative loss function that considers the embeddings of activity labels and their super-classes while learning the common semantics of seen and unseen activities. Experiments on both the THUMOS’14 and the Charades datasets show promising performance in terms of detecting unseen activities. Lingling Zhang 0005, Xiaojun Chang, Jun Liu 0002, Minnan Luo, Sen Wang 0001, ZongYuan Ge, Alex Hauptmann 0001 |
CVPR | 6 |
| 2020 | Grounding Visual Concepts for Zero-Shot Event Detection and Event CaptioningabstractThe flourishing of social media platforms requires techniques for understanding the content of media on a large scale. However, state-of-the art video event understanding approaches remain very limited in terms of their ability to deal with data sparsity, semantically unrepresentative event names, and lack of coherence between visual and textual concepts. Accordingly, in this paper, we propose a method of grounding visual concepts for large-scale Multimedia Event Detection (MED) and Multimedia Event Captioning (MEC) in zero-shot setting. More specifically, our framework composes the following: (1) deriving the novel semantic representations of events from their textual descriptions, rather than event names; (2) aggregating the ranks of grounded concepts for MED tasks. A statistical mean-shift outlier rejection model is proposed to remove the outlying concepts which are incorrectly grounded; and (3) defining MEC tasks and augmenting the MEC training set by the videos detected in MED in a zero-shot setting. To the best of our knowledge, this work is the first time to define and solve the MEC task, which is a further step towards understanding video events. We conduct extensive experiments and achieve state-of-the-art performance on the TRECVID MEDTest dataset, as well as our newly proposed TRECVID-MEC dataset. Zhihui Li 0001, Xiaojun Chang, Lina Yao 0001, Shirui Pan, ZongYuan Ge, Huaxiang Zhang 0001 |
KDD | 5 |
| 2020 | Improve Unseen Domain Generalization via Enhanced Local Color Transformation
Jianhao Xiong, Andre Wang He, Congxin Liu, ZongYuan Ge |
MICCAI (2) | 8 |
| 2020 | Hierarchical Neural Architecture Search for Deep Stereo MatchingabstractTo reduce the human efforts in neural network design, Neural Architecture Search (NAS) has been applied with remarkable success to various high-level vision tasks such as classification and semantic segmentation. The underlying idea for the NAS algorithm is straightforward, namely, to allow the network the ability to choose among a set of operations (\eg convolution with different filter sizes), one is able to find an optimal architecture that is better adapted to the problem at hand. However, so far the success of NAS has not been enjoyed by low-level geometric vision tasks such as stereo matching. This is partly due to the fact that state-of-the-art deep stereo matching networks, designed by humans, are already sheer in size. Directly applying the NAS to such massive structures is computationally prohibitive based on the currently available mainstream computing resources. In this paper, we propose the first \emph{end-to-end} hierarchical NAS framework for deep stereo matching by incorporating task-specific human knowledge into the neural architecture search framework. Specifically, following the gold standard pipeline for deep stereo matching (\ie, feature extraction -- feature volume construction and dense matching), we optimize the architectures of the entire pipeline jointly. Extensive experiments show that our searched network outperforms all state-of-the-art deep stereo matching architectures and is ranked at the top 1 accuracy on KITTI stereo 2012, 2015, and Middlebury benchmarks, as well as the top 1 on SceneFlow dataset with a substantial improvement on the size of the network and the speed of inference. Code available at https://github.com/XuelianCheng/LEAStereo. Xuelian Cheng, Yiran Zhong, Mehrtash Harandi, Yuchao Dai, Xiaojun Chang, Hongdong Li, Tom Drummond, ZongYuan Ge |
NeurIPS | 8 |
| 2020 | Differentiable Neural Architecture Search in Equivalent Space with Exploration EnhancementabstractRecent works on One-Shot Neural Architecture Search (NAS) mostly adopt a bilevel optimization scheme to alternatively optimize the supernet weights and architecture parameters after relaxing the discrete search space into a differentiable space. However, the non-negligible incongruence in their relaxation methods is hard to guarantee the differentiable optimization in the continuous space is equivalent to the optimization in the discrete space. Differently, this paper utilizes a variational graph autoencoder to injectively transform the discrete architecture space into an equivalently continuous latent space, to resolve the incongruence. A probabilistic exploration enhancement method is accordingly devised to encourage intelligent exploration during the architecture search in the latent space, to avoid local optimal in architecture search. As the catastrophic forgetting in differentiable One-Shot NAS deteriorates supernet predictive ability and makes the bilevel optimization inefficient, this paper further proposes an architecture complementation method to relieve this deficiency. We analyze the effectiveness of the proposed method, and a series of experiments have been conducted to compare the proposed method with state-of-the-art One-Shot NAS methods. Miao Zhang 0022, Huiqi Li, Shirui Pan, Xiaojun Chang, ZongYuan Ge, Steven W. Su |
NeurIPS | 5 |
| 2020 | PeNGaRoo, a combined gradient boosting and ensemble learning framework for predicting non-classical secreted proteinsabstractMOTIVATION: Gram-positive bacteria have developed secretion systems to transport proteins across their cell wall, a process that plays an important role during host infection. These secretion mechanisms have also been harnessed for therapeutic purposes in many biotechnology applications. Accordingly, the identification of features that select a protein for efficient secretion from these microorganisms has become an important task. Among all the secreted proteins, 'non-classical' secreted proteins are difficult to identify as they lack discernable signal peptide sequences and can make use of diverse secretion pathways. Currently, several computational methods have been developed to facilitate the discovery of such non-classical secreted proteins; however, the existing methods are based on either simulated or limited experimental datasets. In addition, they often employ basic features to train the models in a simple and coarse-grained manner. The availability of more experimentally validated datasets, advanced feature engineering techniques and novel machine learning approaches creates new opportunities for the development of improved predictors of 'non-classical' secreted proteins from sequence data. RESULTS: In this work, we first constructed a high-quality dataset of experimentally verified 'non-classical' secreted proteins, which we then used to create benchmark datasets. Using these benchmark datasets, we comprehensively analyzed a wide range of features and assessed their individual performance. Subsequently, we developed a two-layer Light Gradient Boosting Machine (LightGBM) ensemble model that integrates several single feature-based models into an overall prediction framework. At this stage, LightGBM, a gradient boosting machine, was used as a machine learning approach and the necessary parameter optimization was performed by a particle swarm optimization strategy. All single feature-based LightGBM models were then integrated into a unified ensemble model to further improve the predictive performance. Consequently, the final ensemble model achieved a superior performance with an accuracy of 0.900, an F-value of 0.903, Matthew's correlation coefficient of 0.803 and an area under the curve value of 0.963, and outperforming previous state-of-the-art predictors on the independent test. Based on our proposed optimal ensemble model, we further developed an accessible online predictor, PeNGaRoo, to serve users' demands. We believe this online web server, together with our proposed methodology, will expedite the discovery of non-classically secreted effector proteins in Gram-positive bacteria and further inspire the development of next-generation predictors. AVAILABILITY AND IMPLEMENTATION: http://pengaroo.erc.monash.edu/. SUPPLEMENTARY INFORMATION: Supplementary data are available at Bioinformatics online. Yanju Zhang, Sha Yu, Ruopeng Xie, Jiahui Li 0007, André Leier, Tatiana T. Marquez-Lago, Tatsuya Akutsu, Alexander Ian Smith, ZongYuan Ge, Jiawei Wang 0002, Trevor Lithgow, Jiangning Song |
Bioinform. | 9 |
| 2020 | Skin lesion segmentation via generative adversarial networks with dual discriminators
Bai Ying Lei, Zaimin Xia, Xudong Jiang 0001, ZongYuan Ge, Yanwu Xu 0001, Jie Du 0001, Siping Chen, Tianfu Wang 0001, Shuqiang Wang |
Medical Image Anal. | 5 |
| 2020 | Improving multi-label chest X-ray disease diagnosis by exploiting disease and health labels dependencies
ZongYuan Ge, Dwarikanath Mahapatra, Xiaojun Chang, Zetao Chen, Lianhua Chi, Huimin Lu 0001 |
Multim. Tools Appl. | 1 |
| 2020 | Training data independent image registration using generative adversarial networks and domain adaptation
Dwarikanath Mahapatra, ZongYuan Ge |
Pattern Recognit. | 2 |
| 2020 | Progressive Transfer Learning and Adversarial Domain Adaptation for Cross-Domain Skin Disease ClassificationabstractDeep learning has been used to analyze and diagnose various skin diseases through medical imaging. However, recent researches show that a well-trained deep learning model may not generalize well to data from different cohorts due to domain shift. Simple data fusion techniques such as combining disease samples from different data sources are not effective to solve this problem. In this paper, we present two methods for a novel task of cross-domain skin disease recognition. Starting from a fully supervised deep convolutional neural network classifier pre-trained on ImageNet, we explore a two-step progressive transfer learning technique by fine-tuning the network on two skin disease datasets. We then propose to adopt adversarial learning as a domain adaptation technique to perform invariant attribute translation from source to target domain in order to improve the recognition performance. In order to evaluate these two methods, we analyze generalization capability of the trained model on melanoma detection, cancer detection, and cross-modality learning tasks on two skin image datasets collected from different clinical settings and cohorts with different disease distributions. The experiments prove the effectiveness of our method in solving the domain shift problem. Yanyang Gu, ZongYuan Ge, C. Paul Bonnington, Jun Zhou 0001 |
IEEE J. Biomed. Health Informatics | 2 |
| 2019 | Model-less Active Compliance for Continuum Robots using Recurrent Neural NetworksabstractEndowing continuum robots with compliance while it interacts with the internal environment of the human body is essential to prevent damage to the robot and the surrounding tissues. Compared with passive compliance, active compliance has the advantages in terms of increasing the force transmission ability and improving safety with monitored force output. Previous studies have demonstrated that active compliance can be achieved based on a complex model of the mechanics combined with a traditional machine learning technique such as a support vector machine. This paper proposes a recurrent neural network (RNN) based approach that avoids the complexity of modeling while capturing nonlinear factors such as hysteresis, friction and delay of the electronics that are not easy to model. The approach is tested on a 3-tendon single-segment continuum robot with force sensors on each cable. Experiments are conducted to demonstrate that the continuum robot with an RNN based feed-forward controller is capable of responding to external forces quickly and entering an unknown environment compliantly. David Jakes, ZongYuan Ge, Liao Wu |
IROS | 2 |
| 2019 | Improving Deep Lesion Detection Using 3D Contextual and Spatial Attention
Qingyi Tao, ZongYuan Ge, Jianfei Cai 0001, Jianxiong Yin, Simon See |
MICCAI (6) | 2 |
| 2019 | Retinal Abnormalities Recognition Using Regional Multitask Learning
Xin Wang 0094, Lie Ju, ZongYuan Ge |
MICCAI (1) | 4 |
| 2019 | Adversarial Pulmonary Pathology Translation for Pairwise Chest X-Ray Data Augmentation
Yunyan Xing, ZongYuan Ge, Dwarikanath Mahapatra, Jarrel Seah, Meng Law, Tom Drummond |
MICCAI (6) | 2 |
| 2019 | Knowledge driven temporal activity localization
Zhihui Li 0001, ZongYuan Ge, Mingjie Li 0006 |
J. Vis. Commun. Image Represent. | 3 |
| 2018 | Structured deep hashing with convolutional neural networks for fast person re-identification
Lin Wu 0001, Yang Wang 0023, ZongYuan Ge, Qichang Hu, Xue Li 0001 |
Comput. Vis. Image Underst. | 3 |
| 2018 | Editorial: Artificial Intelligence for Mobile Robotic Networks
Huimin Lu 0001, Li He 0001, Quan Zhou 0004, ZongYuan Ge |
Mob. Networks Appl. | 4 |
| 2018 | FDCNet: filtering deep convolutional network for marine organism classification
Huimin Lu 0001, Yujie Li 0001, Tomoki Uemura, ZongYuan Ge, Xing Xu 0001, Li He 0001, Seiichi Serikawa, Hyoungseop Kim |
Multim. Tools Appl. | 4 |
| 2017 | Generative OpenMax for Multi-Class Open Set Classification
ZongYuan Ge, Sergey Demyanov, Rahil Garnavi |
BMVC | 1 |
| 2017 | Skin Disease Recognition Using Deep Saliency Features and Multimodal Learning of Dermoscopy and Clinical Images
ZongYuan Ge, Sergey Demyanov, Rajib Chakravorty, Adrian Bowling, Rahil Garnavi |
MICCAI (3) | 1 |
| 2016 | Simple online and realtime trackingabstractThis paper explores a pragmatic approach to multiple object tracking where the main focus is to associate objects efficiently for online and realtime applications. To this end, detection quality is identified as a key factor influencing tracking performance, where changing the detector can improve tracking by up to 18.9%. Despite only using a rudimentary combination of familiar techniques such as the Kalman Filter and Hungarian algorithm for the tracking components, this approach achieves an accuracy comparable to state-of-the-art online trackers. Furthermore, due to the simplicity of our tracking method, the tracker updates at a rate of 260 Hz which is over 20x faster than other state-of-the-art trackers. Alex Bewley, ZongYuan Ge, Lionel Ott, Fabio Ramos 0001, Ben Upcroft |
ICIP | 2 |
| 2016 | Fine-grained classification via mixture of deep convolutional neural networksabstractWe present a novel deep convolutional neural network (DCNN) system for fine-grained image classification, called a mixture of DCNNs (MixDCNN). The fine-grained image classification problem is characterised by large intra-class variations and small inter-class variations. To overcome these problems our proposed MixDCNN system partitions images into K subsets of similar images and learns an expert DCNN for each subset. The output from each of the K DCNNs is combined to form a single classification decision. In contrast to previous techniques, we provide a formulation to perform joint end-to-end training of the K DCNNs simultaneously. Extensive experiments, on three datasets using two network structures (AlexNet and GoogLeNet), show that the proposed MixDCNN system consistently outperforms other methods. It provides a relative improvement of 12.7% and achieves state-of-the-art results on two datasets. ZongYuan Ge, Alex Bewley, Chris McCool, Peter I. Corke, Ben Upcroft, Conrad Sanderson |
WACV | 1 |
| 2015 | Fine-grained bird species recognition via hierarchical subset learningabstractWe propose a novel method to improve fine-grained bird species classification based on hierarchical subset learning. We first form a similarity tree where classes with strong visual correlations are grouped into subsets. An expert local classifier with strong discriminative power to distinguish visually similar classes is then learnt for each subset. On the challenging Caltech200-2011 bird dataset we show that using the hierarchical approach with features derived from a deep convolutional neural network leads to the average accuracy improving from 64.5% to 72.7%, a relative improvement of 12.7%. ZongYuan Ge, Chris McCool, Conrad Sanderson, Alex Bewley, Zetao Chen, Peter I. Corke |
ICIP | 1 |
| 2015 | Modelling local deep convolutional neural network features to improve fine-grained image classificationabstractWe propose a local modelling approach using deep convolutional neural networks (CNNs) for fine-grained image classification. Recently, deep CNNs trained from large datasets have considerably improved the performance of object recognition. However, to date there has been limited work using these deep CNNs as local feature extractors. This partly stems from CNNs having internal representations which are high dimensional, thereby making such representations difficult to model using stochastic models. To overcome this issue, we propose to reduce the dimensionality of one of the internal fully connected layers, in conjunction with layer-restricted retraining to avoid retraining the entire network. The distribution of low-dimensional features obtained from the modified layer is then modelled using a Gaussian mixture model. Comparative experiments show that considerable performance improvements can be achieved on the challenging Fish and UEC FOOD-100 datasets. ZongYuan Ge, Chris McCool, Conrad Sanderson, Peter I. Corke |
ICIP | 1 |
| 2015 | Distance metric learning for feature-agnostic place recognitionabstractThe recent focus on performing visual navigation and place recognition in changing environments has resulted in a large number of heterogeneous techniques each utilizing their own learnt or hand crafted visual features. This paper presents a generally applicable method for learning the appropriate distance metric by which to compare feature responses from any of these techniques in order to perform place recognition under changing environmental conditions. We implement an approach which learns to cluster images captured at spatially proximal locations under different conditions, separated from frames captured at different places. The formulation is a convex optimization, guaranteeing the existence of a global solution. We evaluate the general applicability of our method on two benchmark change datasets using three typical image pre-processing and feature types: GIST, Principal Component Analysis and learnt Convolutional Neural Network features. The results demonstrate that the distance metric learning approach uniformly improves single-image-based visual place recognition performance across all feature types. Furthermore, we demonstrate that this performance improvement is maintained when the sequence-based algorithm SeqSLAM is applied to the single-image place recognition results, leading to state-of-the-art performance. Zetao Chen, Stephanie M. Lowry, Adam Jacobson, ZongYuan Ge, Michael Milford |
IROS | 4 |
| 2014 | Local inter-session variability modelling for object classificationabstractObject classification is plagued by the issue of session variation. Session variation describes any variation that makes one instance of an object look different to another, for instance due to pose or illumination variation. Recent work in the challenging task of face verification has shown that session variability modelling provides a mechanism to overcome some of these limitations. However, for computer vision purposes, it has only been applied in the limited setting of face verification. In this paper we propose a local region based intersession variability (ISV) modelling approach, and apply it to challenging real-world data. We propose a region based session variability modelling approach so that local session variations can be modelled, termed Local ISV. We then demonstrate the efficacy of this technique on a challenging real-world fish image database which includes images taken underwater, providing significant real-world session variations. This Local ISV approach provides a relative performance improvement of, on average, 23% on the challenging MOBIO, Multi-PIE and SCface face databases. It also provides a relative performance improvement of 35% on our challenging fish image dataset. Kaneswaran Anantharajah, ZongYuan Ge, Chris McCool, Simon Denman, Clinton Fookes, Peter I. Corke, Dian Tjondronegoro, Sridha Sridharan |
WACV | 2 |