Weijia Zhang

Dr. Weijia Zhang


Research Focus & Metrics

Academic Career Length

4Years (2023-Present)

(Calculated once Scilit publication data is synced: current year − year of first publication + 1)

Representative Publications

View All Publications
10 December 2025 in AI
Verified
Journal Article
Open access
PDF file
Mohammed Ghazai H AlsubaieMohammed Ghazai H Alsubaie
Mohammed Ghazai H Alsubaie
Suhuai LuoSuhuai Luo
Suhuai Luo
Kamran ShaukatKamran Shaukat
Kamran Shaukat
WZWeijia Zhang
Weijia Zhang
JLJiaming Li
Jiaming Li
Alzheimer’s disease (AD) affects over 50 million people worldwide, making early and accurate diagnosis essential for effective treatment and care planning. Diagnosing AD through neuroimaging continues to face challenges, including reliance on subjective clinical evaluations, the need for manual feature extraction, and limited generalisability across diverse populations. Recent advances in deep learning, especially convolutional neural networks (CNNs) and vision transformers, have improved diagnostic performance, but many models still depend on large labelled datasets and high computational resources. This study introduces an attention-enhanced CNN with a multi-activation fusion (MAF) module and evaluates it using the Alzheimer’s Disease Neuroimaging Initiative dataset. The channel attention mechanism helps the model focus on the most important brain regions in 3D MRI scans, while the MAF module, inspired by multi-head attention, uses parallel fully connected layers with different activation functions to capture varied and complementary feature patterns. This design improves feature representation and increases robustness across heterogeneous patient groups. The proposed model achieved 92.1% accuracy and 0.99 AUC, with precision, recall, and F1-scores of 91.3%, 89.3%, and 92%, respectively. Ten-fold cross-validation confirmed its reliability, showing consistent performance with 91.23% accuracy, 0.93 AUC, 90.29% precision, and 88.30% recall. Comparative analysis also shows that the model outperforms several state-of-the-art deep learning approaches for AD classification. Overall, these findings highlight the potential of combining attention mechanisms with multi-activation modules to improve automated AD diagnosis and enhance diagnostic reliability.

11 April 2025 in Proceedings of the AAAI Conference on Artificial Intelligence
Verified
Journal Article
Open access
PDF file
YWYizhi Wang
Yizhi Wang
WZWeijia Zhang
Weijia Zhang
MZMin-Ling Zhang
Min-Ling Zhang
Partial label learning (PLL) addresses situations where each training example is associated with a set of candidate labels, among which only one corresponds to the true class label. As the candidate labels often come from crowdsourced workers, their generation is inherently dependent on the features of the instance. Existing PLL methods primarily aim to resolve these ambiguous labels to enhance classification accuracy, overlooking the opportunity to use this feature dependency for causal representation learning. This focus on accuracy can make PLL systems vulnerable to stylistic variations and shifts in domain. In this paper, we explore the learning of causal representations within an instance-dependent PLL framework, introducing a new approach that uncovers identifiable latent representations. By separating content from style in the identified causal representation, we introduce CausalPLL+, an algorithm for instance-dependent PLL based on causal representation. Our algorithm performs exceptionally well in terms of both classification accuracy and generalization robustness. Qualitative and quantitative experiments on instance-dependent PLL benchmarks and domain generalization tasks verify the effectiveness of our approach.

1 August 2024
Verified
Conference Paper
Open access
PDF file
WTWei Tang
Wei Tang
WZWeijia Zhang
Weijia Zhang
MZMin-Ling Zhang
Min-Ling Zhang
Tabular data is widely adopted in various machine learning tasks. Current tabular data learning mainly focuses on closed environments, while in real-world applications, open environments are often encountered, where distribution shifts and feature decrements occur, leading to severe performance degradation. Previous studies have primarily focused on addressing distribution shifts, while feature decrements, a unique challenge in tabular data learning, have received relatively little attention. In this paper, we present the first comprehensive study on the problem of Fully Test-Time Adaptation for Feature Decrement in Tabular Data. Through empirical analysis, we identify the suboptimality of existing missing-feature imputation methods and the limited applicability of missing-feature adaptation approaches. To address these challenges, we propose a novel method, LLM-IMPUTE, which leverages Large Language Models (LLMs) to impute missing features without relying on training data. Furthermore, we introduce Augmented-Training LLM (ATLLM), a method designed to enhance the robustness of feature decrements by simulating feature-decrement scenarios during the training phase to address tasks that can not be imputed by LLM-IMPUTE. Extensive experimental results demonstrate that our proposal significantly improves both performance and robustness in missing feature imputation and adaptation scenarios.

28 September 2023 in arXiv
Verified
Preprint
Open access
PDF file
WTWei Tang
Wei Tang
WZWeijia Zhang
Weijia Zhang
MZMin-Ling Zhang
Min-Ling Zhang
In many real-world tasks, the concerned objects can be represented as a multi-instance bag associated with a candidate label set, which consists of one ground-truth label and several false positive labels. Multi-instance partial-label learning (MIPL) is a learning paradigm to deal with such tasks and has achieved favorable performances. Existing MIPL approach follows the instance-space paradigm by assigning augmented candidate label sets of bags to each instance and aggregating bag-level labels from instance-level labels. However, this scheme may be suboptimal as global bag-level information is ignored and the predicted labels of bags are sensitive to predictions of negative instances. In this paper, we study an alternative scheme where a multi-instance bag is embedded into a single vector representation. Accordingly, an intuitive algorithm named DEMIPL, i.e., Disambiguated attention Embedding for Multi-Instance Partial-Label learning, is proposed. DEMIPL employs a disambiguation attention mechanism to aggregate a multi-instance bag into a single vector representation, followed by a momentum-based disambiguation strategy to identify the ground-truth label from the candidate label set. Furthermore, we introduce a real-world MIPL dataset for colorectal cancer classification. Experimental results on benchmark and real-world datasets validate the superiority of DEMIPL against the compared MIPL and partial-label learning approaches.

19 December 2025 in arXiv
Verified
Preprint
Open access
PDF file
WTWei Tang
Wei Tang
YYYin-Fang Yang
Yin-Fang Yang
WZWeijia Zhang
Weijia Zhang
MZMin-Ling Zhang
Min-Ling Zhang
Multi-instance partial-label learning (MIPL) is a weakly supervised framework that extends the principles of multi-instance learning (MIL) and partial-label learning (PLL) to address the challenges of inexact supervision in both instance and label spaces. However, existing MIPL approaches often suffer from poor calibration, undermining classifier reliability. In this work, we propose a plug-and-play calibratable disambiguation loss (CDL) that simultaneously improves classification accuracy and calibration performance. The loss has two instantiations: the first one calibrates predictions based on probabilities from the candidate label set, while the second one integrates probabilities from both candidate and non-candidate label sets. The proposed CDL can be seamlessly incorporated into existing MIPL and PLL frameworks. We provide a theoretical analysis that establishes the lower bound and regularization properties of CDL, demonstrating its superiority over conventional disambiguation losses. Experimental results on benchmark and real-world datasets confirm that our CDL significantly enhances both classification and calibration performance.

14 May 2023 in arXiv
Verified
Preprint
Open access
PDF file
HDHanwen Deng
Hanwen Deng
WZWeijia Zhang
Weijia Zhang
MZMin-Ling Zhang
Min-Ling Zhang
Label noise widely exists in large-scale datasets and significantly degenerates the performances of deep learning algorithms. Due to the non-identifiability of the instance-dependent noise transition matrix, most existing algorithms address the problem by assuming the noisy label generation process to be independent of the instance features. Unfortunately, noisy labels in real-world applications often depend on both the true label and the features. In this work, we tackle instance-dependent label noise with a novel deep generative model that avoids explicitly modeling the noise transition matrix. Our algorithm leverages casual representation learning and simultaneously identifies the high-level content and style latent factors from the data. By exploiting the supervision information of noisy labels with structural causal models, our empirical evaluations on a wide range of synthetic and real-world instance-dependent label noise datasets demonstrate that the proposed algorithm significantly outperforms the state-of-the-art counterparts.