Boyu Wang 0004

dblp:41/6565-4 · DBLP profile ↗
← Back
86ranked-venue papers
5as first author
76since 2021 · last 2026
—ORCID · conflict

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

Artificial intelligence and machine learning · 64 · 4 first-author · 57 since 2021Graphics, computer vision, multimedia, augmented reality and games · 18 · 17 since 2021Databases, data management, data science and information retrieval · 9 · 8 since 2021Applied, interdisciplinary, general and emerging computing · 9 · 1 first-author · 7 since 2021Computer networks · 3 · 2 since 2021Human-computer interaction and ubiquitous computing · 2 · 1 first-author · 1 since 2021
YearPublicationVenuePosition
2026 Graph Domain Adaptation via Homophily-Agnostic Reconstructing Structure
abstract
Graph Domain Adaptation (GDA) transfers knowledge from labeled source graphs to unlabeled target graphs, addressing the challenge of label scarcity. However, existing GDA methods typically assume that both source and target graphs exhibit homophily, leading existing methods to perform poorly when heterophily is present. Furthermore, the lack of labels in the target graph makes it impossible to assess its homophily level beforehand. To address this challenge, we propose a novel homophily-agnostic approach that effectively transfers knowledge between graphs with varying degrees of homophily. Specifically, we adopt a divide-and-conquer strategy that first separately reconstructs highly homophilic and heterophilic variants of both the source and target graphs, and then performs knowledge alignment separately between corresponding graph variants. Extensive experiments conducted on five benchmark datasets demonstrate the superior performance of our approach, particularly highlighting its substantial advantages on heterophilic graphs.
Ruiyi Fang, Ruizhi Pu, Qiuhao Zeng, Hao Zheng 0009, Jiale Cai, Zhimin Mei, Charles Ling 0001, Boyu Wang 0004
AAAI11
2026 Intra-Class Unbiased Prototype Aggregation and Classifier Collaboration for Personalized Federated Learning
abstract
Prototype-based personalized federated learning methods have emerged as a promising strategy due to their ability to represent client-specific class characteristics effectively through learned class prototypes. These prototypes capture salient features of client-local data, facilitating personalized model adaptation. However, existing prototype-based aggregation strategies predominantly rely on weighted averaging, implicitly assuming prototype consistency across clients. This assumption neglects the intrinsic heterogeneity and non-independent and identically distributed (non-IID) nature of client data, compelling diverse local prototypes to align toward a singular global prototype and consequently causing significant aggregation bias. Motivated by observations from intra-class feature saliency analysis, we identify that clients inherently emphasize distinct feature regions even for the same class. To leverage this intra-class diversity, we introduce FedIC, a novel prototype clustering and collaborative classifier optimization approach. Specifically, FedIC first clusters prototypes based on intra-class similarity to form intra-class prototype subspaces, ensuring that aggregation occurs exclusively within each cluster, thus eliminating the bias stemming from forced global unification. To further exploit the benefits of intra-cluster collaboration, we quantify the combined predictive gains of classifiers from clients within the same cluster as a function of classifier combination weights. This targeted aggregation and collaborative optimization strategy effectively circumvents the bias introduced by global alignment. Extensive experiments under various non-IID settings show that FedIC significantly outperforms existing Prototype-based and Clustered PFL Methods.
Hao Zheng 0009, Shiyu Song, Zhigang Hu 0001, Meiguang Zheng, Liu Yang 0015, Aikun Xu, Rongchang Zhao, Ruizhi Pu, Ruiyi Fang, Boyu Wang 0004
AAAI10
2026 HiFC-GAN: Hierarchical Feature-Constrained GAN for Optical-to-SAR Transfer in SAR Target Classification
abstract
The limited availability of high-quality training data poses a persistent challenge for synthetic aperture radar (SAR) target classification. Existing data augmentation methods mainly adopt a simplistic application of GAN-based style transfer techniques to directly synthesize pseudo-SAR images from optical images. However, our in-depth analysis of this cross-modal conversion reveals that such straightforward strategies primarily focus on transferring high-level semantic information (e.g., target shapes), thus failing to adequately capture the essential low-level features unique to SAR imagery (e.g., scattering textures). To address this inherent trade-off between high-level semantic preservation and low-level feature authenticity, we propose a Hierarchical Feature-Constrained GAN (HiFC-GAN) tailored for optical-to-SAR style transfer. Specifically, HiFC-GAN enhances the representation of low-level SAR features by introducing local texture contrast constraints at shallow layers, while introducing explicit feature mapping constraints at deeper layers to maintain high-level semantic consistency throughout the reconstruction process. Experimental results demonstrate that HiFC-GAN significantly outperforms existing GAN-based techniques in image generation quality, particularly improving the low-level feature authenticity of pseudo-SAR images. Moreover, the generated pseudo-SAR images further improve the performance of downstream target classification tasks, yielding accuracy gains ranging from 3.56% to 5.90% on average with mainstream CNN-based models.
Hao Zheng 0009, Meiguang Zheng, Zhigang Hu 0001, Liu Yang 0015, Aikun Xu, Tingxuan Chen, Rongchang Zhao, Boyu Wang 0004
AAAI8
2026 Multi-VLM collaborated adaptive sampling for enhanced data pruning
Changfan Wang, Wei Xu 0046, Huahui Yi, Kang Li 0004, Boyu Wang 0004, Qicheng Lao
Neurocomputing7
2026 Class-Missing Semi-supervised document key information extraction via synergistic refinement estimation
abstract
Current methods for document key information extraction (DKIE) rely heavily on labeled data with high annotation costs. To mitigate this issue, the semi-supervised learning (SSL) paradigm, which utilizes unlabeled document samples, has gained broad attention in DKIE. However, existing SSL methods require labeled and unlabeled data to share an identical label space, which is impractical in many DKIE tasks (i.e., some unlabeled samples do not belong to any known classes in the labeled set). In this paper, we formulate this problem as Class-Missing Semi-supervised (CMSS) DKIE. In DKIE, unknown classes usually belong to minority and fine-grained categories, intensifying the misconnections between known and unknown classes and making CMSS more challenging. To address this issue, we propose Synergistic Refinement Estimation (SRE), a progressive prototype estimation scheme that alleviates the unknown classes bias to the majority known classes on long-tailed unlabeled data. Furthermore, dynamic threshold hash rectification and structural calibration mechanisms are proposed to correct connections between fine-grained classes. Extensive experimental results demonstrate that SRE surpasses existing state-of-the-art methods on several DKIE benchmarks. Code is available at https://github.com/anonymoulink/SRE_DKIE .
Yonghong Song, Boyu Wang 0004, Yankai Cao, Jiayang Ren, Chaojie Ji, Qi Zhang 0096, Qiangqiang Mao
Inf. Process. Manag.3
2026 Test-Time Domain Adaptation With Time-Frequency Consistency and Instance-Aware Batch Renormalization for Online Machinery Fault Diagnosis
Jian Zhu 0001, Bairui Long, Lunke Fei, Yutang Xiao, Boyu Wang 0004, Ruichu Cai
IEEE Trans Autom. Sci. Eng.5
2026 Dual-Branch Attention-Based Frequency Domain Network for Cross-Subject SSVEP-BCIs
abstract
Steady-state visual evoked potential-based brain-computer interfaces (SSVEP-BCIs) hold significant promise for enabling high-speed human-computer interaction in real-world scenarios. However, existing frequency-domain decoding methods treat frequency spectrum features (the real and imaginary spectrum features) as a single feature without considering their unique spatial and spectral characteristics, resulting in insufficient generalizable features and limited classification accuracy in cross-subject scenarios. To address this issue, we propose a Dual-Branch Attention-Based Frequency Domain Network (DB-AFDNet) to independently decode real and imaginary spectral components, aiming to acquire more discriminative and generalizable features for cross-subject applications. Specifically, we construct inter-branch attention similarity constraints to encourage the two branches to have similar attention properties, promoting to learn the consensus characteristics in the dual branches. Furthermore, we propose intra-branch orthogonality constraints to explore branch-specific discriminative features to learn generalizable features. Experimental studies on two public datasets, the Benchmark and Beta datasets, demonstrate that DB-AFDNet outperforms state-of-the-art methods in cross-subject classification, achieving a relative improvement of 1.36$\%$ and 1.45$\%$, respectively.
Yi Yang 0067, Ze Wang 0001, Ziyu Jia, Boyu Wang 0004, Shangen Zhang, Chiman Wong, Xiaorong Gao, Tzyy-Ping Jung, Feng Wan 0003
IEEE J. Biomed. Health Informatics4
2025 Leveraging Group Classification with Descending Soft Labeling for Deep Imbalanced Regression
abstract
Deep imbalanced regression (DIR), where the target values have a highly skewed distribution and are also continuous, is an intriguing yet under-explored problem in machine learning. While recent works have already shown that incorporating various classification-based regularizers can produce enhanced outcomes, the role of classification remains elusive in DIR. Moreover, such regularizers (e.g., contrastive penalties) merely focus on learning discriminative features of data, which inevitably results in ignorance of either continuity or similarity across the data. To address these issues, we first bridge the connection between the objectives of DIR and classification from a Bayesian perspective. Consequently, this motivates us to decompose the objective of DIR into a combination of classification and regression tasks, which naturally guides us toward a divide-and-conquer manner to solve the DIR problem. Specifically, by aggregating the data at nearby labels into the same groups, we introduce an ordinal group-aware contrastive learning loss along with a multi-experts regressor to tackle the different groups of data thereby maintaining the data continuity. Meanwhile, considering the similarity between the groups, we also propose a symmetric descending soft labeling strategy to exploit the intrinsic similarity across the data, which allows classification to facilitate regression more effectively. Extensive experiments on real-world datasets also validate the effectiveness of our method.
Ruizhi Pu, Gezheng Xu, Ruiyi Fang, Bing-Kun Bao, Charles Ling 0001, Boyu Wang 0004
AAAI6
2025 MABR: Multilayer Adversarial Bias Removal Without Prior Bias Knowledge
abstract
Models trained on real-world data often mirror and exacerbate existing social biases. Traditional methods for mitigating these biases typically require prior knowledge of the specific biases to be addressed, and the social groups associated with each instance. In this paper, we introduce a novel adversarial training strategy that operates withour relying on prior bias-type knowledge (e.g., gender or racial bias) and protected attribute labels. Our approach dynamically identifies biases during model training by utilizing auxiliary bias detector. These detected biases are simultaneously mitigated through adversarial training. Crucially, we implement these bias detectors at various levels of the feature maps of the main model, enabling the detection of a broader and more nuanced range of bias features. Through experiments on racial and gender biases in sentiment and occupation classification tasks, our method effectively reduces social biases without the need for demographic annotations. Moreover, our approach not only matches but often surpasses the efficacy of methods that require detailed demographic insights, marking a significant advancement in bias mitigation techniques.
Maxwell J. Yin, Boyu Wang 0004, Charles Ling 0001
AAAI2
2025 ConFREE: Conflict-free Client Update Aggregation for Personalized Federated Learning
abstract
Negative transfer (NF) is a critical challenge in personalized federated learning (pFL). Existing methods primarily focus on adapting local data distribution on the client side, which can only resist NF, rather than avoid NF itself. To tackle NF at its root, we investigate its mechanism through the lens of the global model, and argue that it is caused by update conflicts among clients during server aggregation. In light of this, we propose a conflict-free client update aggregation strategy (ConFREE), which enables us to avoid NF in pFL. Specifically, ConFREE guides the global update direction by constructing a conflict-free guidance vector through projection and utilizes the optimal local improvements of the worst-performing clients near the guidance vector to regularize server aggregation. This prevents the conflicting components of updates from transferring, achieving balanced updates across different clients. Notably, ConFREE is model-agnostic and can be straightforwardly adopted as a complement to enhance various existing NF-resistance methods implemented on the client side. Extensive experiments demonstrate substantial improvements to existing pFL algorithms by leveraging ConFREE.
Hao Zheng 0009, Zhigang Hu 0001, Liu Yang 0015, Meiguang Zheng, Aikun Xu, Boyu Wang 0004
AAAI6
2025 Active Learning for Lesion Segmentation Using Contrastive Learning with Strong Augmentation
abstract
Active learning effectively reduces annotation costs while enhancing model performance in medical image segmentation tasks. One-shot active learning presents a highly practical scenario where valuable samples for annotation are selected in a single round. However, current one-shot active learning methods predominantly rely on sampling selection methods based on global information. In contrast, focusing on the selection of features specific to local lesion regions would be more targeted and effective. In this work, we introduce a novel deep active learning framework specifically designed for pathological lesion segmentation tasks. To enable the model to effectively capture lesion-related regions of interest, we propose a strong augmentation strategy for image samples in self-supervised contrastive training. These strong augmentation samples are generated through cluster-based background subtraction using cluster projector, thereby emphasizing the features of the target lesion and improving the model's sensitivity to these areas. We utilize the Segment Anything Model as the base model to facilitate training and sample selection in a one-shot manner. Experimental results on two lesion segmentation datasets demonstrate that the proposed framework outperforms several existing active learning methods.
Jianyuan Li, Xiong Luo, Boyu Wang 0004
BIBM3
2025 Unveil Inversion and Invariance in Flow Transformer for Versatile Image Editing
abstract
Leveraging the large generative prior of the flow transformer for tuning-free image editing requires authentic inversion to project the image into the model’s domain and a flexible invariance control mechanism to preserve non-target contents. However, the prevailing diffusion inversion performs deficiently in flow-based models, and the invariance control cannot reconcile diverse rigid and non-rigid editing tasks. To address these, we systematically analyze the inversion and invariance control based on the flow transformer. Specifically, we unveil that the Euler inversion shares a similar structure to DDIM yet is more susceptible to the approximation error. Thus, we propose a two-stage inversion to first refine the velocity estimation and then compensate for the leftover error, which pivots closely to the model prior and benefits editing. Meanwhile, we propose the invariance control that manipulates the text features within the adaptive layer normalization, connecting the changes in the text prompt to image semantics. This mechanism can simultaneously preserve the non-target contents while allowing rigid and non-rigid manipulation, enabling a wide range of editing types such as visual text, quantity, facial expression, etc. Experiments on versatile scenarios validate that our framework achieves flexible and accurate editing, unlocking the potential of the flow transformer for versatile image editing. Project Page is here.
Pengcheng Xu 0008, Boyuan Jiang, Xiaobin Hu, Donghao Luo 0001, Qingdong He, Jiangning Zhang, Chengjie Wang 0001, Yunsheng Wu, Charles Ling 0001, Boyu Wang 0004
CVPR10
2025 FedCALM: Conflict-aware Layer-wise Mitigation for Selective Aggregation in Deeper Personalized Federated Learning
abstract
Server aggregation conflict is a key challenge in personalized federated learning (PFL). While existing PFL methods have achieved significant progress with shallow base models (e.g., four-layer CNNs), they often overlook the negative impacts of deeper base models on personalization mechanisms. In this paper, we identify the phenomenon of deep model degradation in PFL, where as base model depth increases, the model becomes more sensitive to local client data distributions, thereby exacerbating server aggregation conflicts and ultimately reducing overall model performance. Moreover, we show that these conflicts manifest in insufficient global average updates and mutual constraints between clients. Motivated by our analysis, we proposed a two-stage conflict-aware layer-wise mitigation algorithm (FedCALM), which first constructs a conflict-free global update to alleviate negative conflicts, and then maximizes the benefits of all clients through a conflict-aware strategy. Notably, our method naturally leads to a selective mechanism that balances the tradeoff between clients involved in aggregation and the tolerance for conflicts. Consequently, it can boost the positive contribution to the clients even with the greatest conflicts with the global update. Extensive experiments across multiple datasets and deeper base models demonstrate that FedCALM outperforms four state-of-the-art (SOTA) methods by up to 9.88% and seamlessly integrates into existing PFL methods with performance improvements of up to 9.01%.
Hao Zheng 0009, Zhigang Hu 0001, Liu Yang 0015, Meiguang Zheng, Aikun Xu, Boyu Wang 0004
CVPR6
2025 On the Benefits of Attribute-Driven Graph Domain Adaptation
abstract
Graph Domain Adaptation (GDA) addresses a pressing challenge in cross-network learning, particularly pertinent due to the absence of labeled data in real-world graph datasets. Recent studies attempted to learn domain invariant representations by eliminating structural shifts between graphs. In this work, we show that existing methodologies have overlooked the significance of the graph node attribute, a pivotal factor for graph domain alignment. Specifically, we first reveal the impact of node attributes for GDA by theoretically proving that in addition to the graph structural divergence between the domains, the node attribute discrepancy also plays a critical role in GDA. Moreover, we also empirically show that the attribute shift is more substantial than the topology shift, which further underscore the importance of node attribute alignment in GDA. Inspired by this finding, a novel cross-channel module is developed to fuse and align both views between the source and target graphs for GDA. Experimental results on a variety of benchmark verify the effectiveness of our method.
Ruiyi Fang, Bingheng Li, Zhao Kang 0001, Qiuhao Zeng, Nima Hosseini Dashtbayaz, Ruizhi Pu, Charles Ling 0001, Boyu Wang 0004
ICLR8
2025 Event-Driven Online Vertical Federated Learning
abstract
Online learning is more adaptable to real-world scenarios in Vertical Federated Learning (VFL) compared to offline learning. However, integrating online learning into VFL presents challenges due to the unique nature of VFL, where clients possess non-intersecting feature sets for the same sample. In real-world scenarios, the clients may not receive data streaming for the disjoint features for the same entity synchronously. Instead, the data are typically generated by an *event* relevant to only a subset of clients. We are the first to identify these challenges in online VFL, which have been overlooked by previous research. To address these challenges, we proposed an event-driven online VFL framework. In this framework, only a subset of clients were activated during each event, while the remaining clients passively collaborated in the learning process. Furthermore, we incorporated *dynamic local regret (DLR)* into VFL to address the challenges posed by online learning problems with non-convex models within a non-stationary environment. We conducted a comprehensive regret analysis of our proposed framework, specifically examining the DLR under non-convex conditions with event-driven online VFL. Extensive experiments demonstrated that our proposed framework was more stable than the existing online VFL framework under non-stationary data conditions while also significantly reducing communication and computation costs.
Ganyu Wang, Boyu Wang 0004, Bin Gu 0001, Charles Ling 0001
ICLR2
2025 ZETA: Leveraging Z-order Curves for Efficient Top-k Attention
abstract
Over recent years, the Transformer has become a fundamental building block for sequence modeling architectures. Yet at its core is the use of self-attention, whose memory and computational cost grow quadratically with the sequence length $N$, rendering it prohibitively expensive for long sequences. A promising approach is top-$k$ attention, which selects only the $k$ most relevant tokens and achieves performance comparable to vanilla self-attention while significantly reducing space and computational demands. However, causal masks require the current query token to only attend to past tokens, preventing existing top-$k$ attention methods from efficiently searching for the most relevant tokens in parallel, thereby limiting training efficiency. In this work, we propose ZETA, leveraging Z-Order Curves for Efficient Top-k Attention, to enable parallel querying of past tokens for entire sequences. We first theoretically show that the choice of key and query dimensions involves a trade-off between the curse of dimensionality and the preservation of relative distances after projection. In light of this insight, we propose reducing the dimensionality of keys and queries in contrast to values and further leveraging Z-order curves to map low-dimensional keys and queries into one-dimensional space, which permits parallel sorting, thereby largely improving the efficiency for top-$k$ token selection. Experimental results demonstrate that ZETA~matches the performance of standard attention on synthetic tasks Associative Recall and outperforms attention and its variants on Long-Range Arena and WikiText-103 language modeling.
Qiuhao Zeng, Jerry Huang, Peng Lu 0006, Gezheng Xu, Boxing Chen, Charles Ling 0001, Boyu Wang 0004
ICLR7
2025 Homophily Enhanced Graph Domain Adaptation
abstract
Graph Domain Adaptation (GDA) transfers knowledge from labeled source graphs to unlabeled target graphs, addressing the challenge of label scarcity. In this paper, we highlight the significance of graph homophily, a pivotal factor for graph domain alignment, which, however, has long been overlooked in existing approaches. Specifically, our analysis first reveals that homophily discrepancies exist in benchmarks. Moreover, we also show that homophily discrepancies degrade GDA performance from both empirical and theoretical aspects, which further underscores the importance of homophily alignment in GDA. Inspired by this finding, we propose a novel homophily alignment algorithm that employs mixed filters to smooth graph signals, thereby effectively capturing and mitigating homophily discrepancies between graphs. Experimental results on a variety of benchmarks verify the effectiveness of our method.
Ruiyi Fang, Bingheng Li, Ruizhi Pu, Qiuhao Zeng, Gezheng Xu, Charles Ling 0001, Boyu Wang 0004
ICML8
2025 FedOne: Query-Efficient Federated Learning for Black-box Discrete Prompt Learning
abstract
Black-Box Discrete Prompt Learning (BDPL) is a prompt-tuning method that optimizes discrete prompts without accessing model parameters or gradients, making the prompt tuning on a cloud-based Large Language Model (LLM) feasible. Adapting Federated Learning (FL) to BDPL could further enhance prompt tuning performance by leveraging data from diverse sources. However, all previous research on federated black-box prompt tuning had neglected the substantial query cost associated with the cloud-based LLM service. To address this gap, we conducted a theoretical analysis of query efficiency within the context of federated black-box prompt tuning. Our findings revealed that degrading FedAvg to activate only one client per round, a strategy we called \textit{FedOne}, enabled optimal query efficiency in federated black-box prompt learning. Building on this insight, we proposed the FedOne framework, a federated black-box discrete prompt learning method designed to maximize query efficiency when interacting with cloud-based LLMs. We conducted numerical experiments on various aspects of our framework, demonstrating a significant improvement in query efficiency, which aligns with our theoretical results.
Ganyu Wang, Jinjie Fang, Maxwell J. Yin, Bin Gu 0001, Xi Chen 0009, Boyu Wang 0004, Yi Chang 0001, Charles Ling 0001
ICML6
2025 Lesion Boundary-Aware Adaptation of Segment Anything Model for 2D Medical Image
abstract
The Segment Anything Model (SAM), serving as a foundational vision model, has demonstrated an extraordinary capability in segmenting natural images. However, its efficacy in the domain of medical image analysis leaves much to be desired, primarily due to the irregular shapes and indistinct edges characteristic of lesions. There is a pressing need to augment SAM’s proficiency in recognizing lesion boundaries. Achieving precise segmentation of such lesions requires a blend of high-level global semantic information and low-level local boundary details. In response to this challenge, we introduce an auxiliary boundary-aware Convolutional Neural Network (CNN) module, equipped with a boundary generator, to enhance the model’s focus on boundary feature extraction. Furthermore, to leverage both the intricate low-level features in the lower layers and the high-level textural features in the deeper layers, we employ feature adapters to fuse the multi-scale features derived from the SAM encoder, thereby aggregating a wealth of enriched information. The performance superiority of our model is demonstrated through comprehensive evaluation on three different medical image segmentation tasks, and experimental results highlight the effectiveness of our proposed model.
Jianyuan Li, Xiong Luo, Boyu Wang 0004
IJCNN3
2025 Versatile Transferable Unlearnable Example Generator
abstract
The rapid growth of publicly available data has fueled deep learning advancements but also raises concerns about unauthorized data usage. Unlearnable Examples (UEs) have emerged as a data protection strategy that introduces imperceptible perturbations to prevent unauthorized learning. However, most existing UE methods produce perturbations strongly tied to specific training sets, leading to a significant drop in unlearnability when applied to unseen data or tasks. In this paper, we argue that for broad applicability, UEs should maintain their effectiveness across diverse application scenarios. To this end, we conduct the first comprehensive study on the transferability of UEs across diverse and practical yet demanding settings. Specifically, we identify key scenarios that pose significant challenges for existing UE methods, including varying styles, out-of-distribution classes, resolutions, and architectures. Moreover, we propose $\textbf{Versatile Transferable Generator}$ (VTG), a transferable generator designed to safeguard data across various conditions. Specifically, VTG integrates Adversarial Domain Augmentation (ADA) into the generator’s training process to synthesize out-of-distribution samples, thereby improving its generalizability to unseen scenarios. Furthermore, we propose a Perturbation-Label Coupling (PLC) mechanism that leverages contrastive learning to directly align perturbations with class labels. This approach reduces the generator’s reliance on data semantics, allowing VTG to produce unlearnable perturbations in a distribution-agnostic manner. Extensive experiments demonstrate the effectiveness and broad applicability of our approach. Code is available at https://github.com/zhli-cs/VTG.
Jiale Cai, Gezheng Xu, Hao Zheng 0009, Qiuyue Li, Fan Zhou 0006, Charles Ling 0001, Boyu Wang 0004
NeurIPS9
2025 A simple remedy for failure modes in physics informed neural networks
Ghazal Farhani, Nima Hosseini Dashtbayaz, Alexander Kazachek, Boyu Wang 0004
Neural Networks4
2025 FedELR: When federated learning meets learning with noisy labels
Ruizhi Pu, Lixing Yu, Shaojie Zhan, Gezheng Xu, Fan Zhou 0006, Charles Ling 0001, Boyu Wang 0004
Neural Networks7
2025 Homeomorphism Prior for False Positive and Negative Problem in Medical Image Dense Contrastive Representation Learning
abstract
Dense contrastive representation learning (DCRL) has greatly improved the learning efficiency for image dense prediction tasks, showing its great potential to reduce the large costs of medical image collection and dense annotation. However, the properties of medical images make unreliable correspondence discovery, bringing an open problem of large-scale false positive and negative (FP&N) pairs in DCRL. In this paper, we propose GEoMetric vIsual deNse sImilarity (GEMINI) learning which embeds the homeomorphism prior to DCRL and enables a reliable correspondence discovery for effective dense contrast. We proposes a deformable homeomorphism learning (DHL) which models the homeomorphism of medical images and learns to estimate a deformable mapping to predict the pixels' correspondence under the condition of topological preservation. It effectively reduces the searching space of pairing and drives an implicit and soft learning of negative pairs via gradient. We also proposes a geometric semantic similarity (GSS) which extracts semantic information in features to measure the alignment degree for the correspondence learning. It will promote the learning efficiency and performance of deformation, constructing positive pairs reliably. We implement two practical variants on two typical representation learning tasks in our experiments. Our promising results on seven datasets which outperform the existing methods show our great superiority. We will release our code at a companion website.
Yuting He 0001, Boyu Wang 0004, Rongjun Ge, Yang Chen 0008, Guanyu Yang 0001, Shuo Li 0001
IEEE Trans. Pattern Anal. Mach. Intell.2
2025 Unraveling the Mysteries of Label Noise in Source-Free Domain Adaptation: Theory and Practice
abstract
Recent source-free domain adaptation (SFDA) methods have focused on learning meaningful cluster structures in feature space, successfully adapting the knowledge from the source domain to the unlabeled target domain without accessing the private source data. However, existing methods rely on pseudo-labels generated by source models that can be noisy due to domain shift, presenting a significant challenge to their efficacy. In this paper, we study SFDA from the perspective of learning with label noise (LLN) and prove that the label noise in SFDA, unlike in conventional LLN scenarios, follows a different distribution assumption. This discrepancy renders some existing LLN methods less effective in SFDA. To address this issue and comprehensively improve adaptation performance, we tackle label noise in SFDA from two perspectives. First, we demonstrate that the early-time training phenomenon (ETP), previously observed in LLN settings, still exists in SFDA. Hence, we introduce a simple yet effective approach to leveraging ETP to improve current SFDA algorithms. Second, we propose a noise and variance control module, mitigating the label noise discrepancy between SFDA and LLN and enhancing the effectiveness of LLN methods in SFDA. Extensive empirical evaluation and analysis of four benchmarks show that our methods substantially outperform existing baselines.
Gezheng Xu, Pengcheng Xu 0008, Jiaqi Li 0005, Ruizhi Pu, Changjian Shui, A. Ian McLeod, Boyu Wang 0004, Charles Ling 0001
IEEE Trans. Pattern Anal. Mach. Intell.8
2025 Exploiting the Intrinsic Neighborhood Semantic Structure for Domain Adaptation in EEG-Based Emotion Recognition
abstract
Due to the inherent non-stationarity and individual differences present in electroencephalogram (EEG) signals, developing a generalizable model that performs well on new subjects is challenging in EEG-based emotion recognition. Most existing domain adaptation (DA) methods typically mitigate these discrepancies by aligning the marginal distributions of domain feature representations. However, when there is a significant difference in the class-conditional distribution between domain features and labels, the domain-invariant features learned by aligning marginal distributions may have limited discriminative ability for unlabeled target instances or even prove counterproductive. To address this issue, we propose a Neighborhood Semantic Aware Learning-based Dynamic Graph Attention Convolution (NSAL-DGAT) approach that learns target semantic information by considering the inter-domain semantic topological structure, thereby improving classifier adaptation for target instances. Specifically, the proposed NSAL framework is designed to capitalize on the insight that after domain feature alignment, some target samples and their neighboring source samples exhibit similar semantics. By leveraging the neighborhood topological structure, we extract and incorporate semantic target features to train a more transferable classifier. Besides, we implement an entropy weighting mechanism to emphasize representative target semantic information, encouraging target instances to prioritize high-confidence individuals within the source neighborhood. We have conducted extensive experiments on the public SEED dataset and our collected the Hearing-Impaired EEG Dataset (HIED). The experimental results underscore the efficacy of our proposed NSAL-DGAT approach, showcasing state-of-the-art accuracy in subject-dependent as well as subject-independent scenarios. The source code is available at https://github.com/YYingDL/NSAL-DGAT.
Yi Yang 0067, Ze Wang 0001, Yu Song 0004, Ziyu Jia, Boyu Wang 0004, Tzyy-Ping Jung, Feng Wan 0003
IEEE Trans. Affect. Comput.5
2024 Generalizing across Temporal Domains with Koopman Operators
abstract
In the field of domain generalization, the task of constructing a predictive model capable of generalizing to a target domain without access to target data remains challenging. This problem becomes further complicated when considering evolving dynamics between domains. While various approaches have been proposed to address this issue, a comprehensive understanding of the underlying generalization theory is still lacking. In this study, we contribute novel theoretic results that aligning conditional distribution leads to the reduction of generalization bounds. Our analysis serves as a key motivation for solving the Temporal Domain Generalization (TDG) problem through the application of Koopman Neural Operators, resulting in Temporal Koopman Networks (TKNets). By employing Koopman Neural Operators, we effectively address the time-evolving distributions encountered in TDG using the principles of Koopman theory, where measurement functions are sought to establish linear transition relations between evolving domains. Through empirical evaluations conducted on synthetic and real-world datasets, we validate the effectiveness of our proposed approach.
Qiuhao Zeng, Wei Wang 0036, Fan Zhou 0006, Gezheng Xu, Ruizhi Pu, Changjian Shui, Christian Gagné 0001, Charles Ling 0001, Boyu Wang 0004
AAAI10
2024 Less or More From Teacher: Exploiting Trilateral Geometry For Knowledge Distillation
abstract
Knowledge distillation aims to train a compact student network using soft supervision from a larger teacher network and hard supervision from ground truths. However, determining an optimal knowledge fusion ratio that balances these supervisory signals remains challenging. Prior methods generally resort to a constant or heuristic-based fusion ratio, which often falls short of a proper balance. In this study, we introduce a novel adaptive method for learning a sample-wise knowledge fusion ratio, exploiting both the correctness of teacher and student, as well as how well the student mimics the teacher on each sample. Our method naturally leads to the \textit{intra-sample} trilateral geometric relations among the student prediction ($\mathcal{S}$), teacher prediction ($\mathcal{T}$), and ground truth ($\mathcal{G}$). To counterbalance the impact of outliers, we further extend to the \textit{inter-sample} relations, incorporating the teacher's global average prediction ($\mathcal{\bar{T}})$ for samples within the same class. A simple neural network then learns the implicit mapping from the intra- and inter-sample relations to an adaptive, sample-wise knowledge fusion ratio in a bilevel-optimization manner. Our approach provides a simple, practical, and adaptable solution for knowledge distillation that can be employed across various architectures and model sizes. Extensive experiments demonstrate consistent improvements over other loss re-weighting methods on image classification, attack detection, and click-through rate prediction.
Chengming Hu, Haolun Wu, Chen Ma 0001, Xi Chen 0009, Boyu Wang 0004, Jun Yan 0007, Xue (Steve) Liu
ICLR6
2024 Latent Trajectory Learning for Limited Timestamps under Distribution Shift over Time
abstract
Distribution shifts over time are common in real-world machine-learning applications. This scenario is formulated as Evolving Domain Generalization (EDG), where models aim to generalize well to unseen target domains in a time-varying system by learning and leveraging the underlying evolving pattern of the distribution shifts across domains. However, existing methods encounter challenges due to the limited number of timestamps (every domain corresponds to a timestamp) in EDG datasets, leading to difficulties in capturing evolving dynamics and risking overfitting to the sparse timestamps, which hampers their generalization and adaptability to new tasks. To address this limitation, we propose a novel approach SDE-EDG that collects the Infinitely Fined-Grid Evolving Trajectory (IFGET) of the data distribution with continuous-interpolated samples to bridge temporal gaps (intervals between two successive timestamps). Furthermore, by leveraging the inherent capacity of Stochastic Differential Equations (SDEs) to capture continuous trajectories, we propose their use to align SDE-modeled trajectories with IFGET across domains, thus enabling the capture of evolving distribution trends. We evaluate our approach on several benchmark datasets and demonstrate that it can achieve superior performance compared to existing state-of-the-art methods.
Qiuhao Zeng, Changjian Shui, Long-Kai Huang, Xi Chen 0009, Charles Ling 0001, Boyu Wang 0004
ICLR7
2024 Intersectional Unfairness Discovery
abstract
AI systems have been shown to produce unfair results for certain subgroups of population, highlighting the need to understand bias on certain sensitive attributes. Current research often falls short, primarily focusing on the subgroups characterized by a single sensitive attribute, while neglecting the nature of intersectional fairness of multiple sensitive attributes. This paper focuses on its one fundamental aspect by discovering diverse high-bias intersectional sensitive attributes. Specifically, we propose a Bias-Guided Generative Network (BGGN). By treating each bias value as a reward, BGGN efficiently generates high-bias intersectional sensitive attributes. Experiments on real-world text and image datasets demonstrate a diverse and efficient discovery of BGGN. To further evaluate the generated unseen but possible unfair intersectional sensitive attributes, we formulate them as prompts and use modern generative AI to produce new text and images. The results of frequently generating biased data provides new insights of discovering potential unfairness in popular modern generative AI systems. Warning: This paper contains examples that are offensive in nature.
Gezheng Xu, Qi Chen 0015, Charles Ling 0001, Boyu Wang 0004, Changjian Shui
ICML4
2024 Physics-Informed Neural Networks: Minimizing Residual Loss with Wide Networks and Effective Activations
Nima Hosseini Dashtbayaz, Ghazal Farhani, Boyu Wang 0004, Charles Ling 0001
IJCAI3
2024 Towards Understanding Evolving Patterns in Sequential Data
abstract
In many machine learning tasks, data is inherently sequential. Most existing algorithms learn from sequential data in an auto-regressive manner, which predicts the next unseen data point based on the observed sequence, implicitly assuming the presence of an \emph{evolving pattern} embedded in the data that can be leveraged. However, identifying and assessing evolving patterns in learning tasks often relies on subjective judgments rooted in the prior knowledge of human experts, lacking a standardized quantitative measure. Furthermore, such measures enable us to determine the suitability of employing sequential models effectively and make informed decisions on the temporal order of time series data, and feature/data selection processes. To address this issue, we introduce the Evolving Rate (EvoRate), which quantitatively approximates the intensity of evolving patterns in the data with Mutual Information. Furthermore, in some temporal data with neural mutual information estimations, we only have snapshots at different timestamps, lacking correspondence, which hinders EvoRate estimation. To tackle this challenge, we propose EvoRate$_\mathcal{W}$, aiming to establish correspondence with optimal transport for estimating the first-order EvoRate. Experiments on synthetic and real-world datasets including images and tabular data validate the efficacy of our EvoRate.
Qiuhao Zeng, Long-Kai Huang, Qi Chen 0015, Charles Ling 0001, Boyu Wang 0004
NeurIPS5
2024 An Efficient Federated Learning Framework for IoT Intrusion Detection
abstract
The exponential growth of the Internet of Things (IoT) ecosystems has raised significant cybersecurity concerns. Deep learning (DL)-based methods have shown promising performance in detecting potential cyber threats in IoT networks. However, as these methods often involve data centralization, they can pose serious data privacy issues for IoT users and increase the communication burden of local networks. Federated learning (FL), as a distributed learning paradigm, enables privacy-preserving training of IoT intrusion detection models by requiring only model updates from IoT devices. However, the resource-constrained nature of IoT devices can significantly decrease FL training efficiencies, such as increased training latency and delayed convergence speed. Moreover, the data heterogeneous issues of IoT devices can also impact the accuracy and robustness of the trained model. To address these challenges, we propose an efficient FL framework, FedKD-Prox, based on federated proximal (FedProx) and knowledge distillation (KD). To improve the prediction accuracy within a limited time budget, the proposed framework aims to efficiently exploit the computation capability of the IoT trainers, reduce the communication overhead of FL, and alleviate the impact of heterogeneous data issues. The simulation results show that FedKD-Prox achieves higher accuracy and improves the robustness of the trained intrusion detection model.
Yushen Chen, Fang Fang 0005, Boyu Wang 0004, Lan Zhang 0005
VTC Fall3
2024 A fast local citation recommendation algorithm scalable to multi-topics
Maxwell J. Yin, Boyu Wang 0004, Charles Ling 0001
Expert Syst. Appl.2
2024 Unified bi-encoder bispace-discriminator disentanglement for cross-domain echocardiography segmentation
Xiaoxiao Cui, Boyu Wang 0004, Shanzhi Jiang, Zhi Liu 0004, Hongji Xu, Li-Zhen Cui 0001, Shuo Li 0001
Knowl. Based Syst.2
2024 Contrastive learning based open-set recognition with unknown score
Yuan Zhou 0006, Songyu Fang, Shuoshi Li, Boyu Wang 0004, Sun-Yuan Kung
Knowl. Based Syst.4
2024 Secure and fast asynchronous Vertical Federated Learning via cascaded hybrid optimization
Ganyu Wang, Xiang Li 0012, Boyu Wang 0004, Bin Gu 0001, Charles Ling 0001
Mach. Learn.4
2024 Source-Free Domain Adaptation for Question Answering with Masked Self-training
abstract
Abstract Previous unsupervised domain adaptation (UDA) methods for question answering (QA) require access to source domain data while fine-tuning the model for the target domain. Source domain data may, however, contain sensitive information and should be protected. In this study, we investigate a more challenging setting, source-free UDA, in which we have only the pretrained source model and target domain data, without access to source domain data. We propose a novel self-training approach to QA models that integrates a specially designed mask module for domain adaptation. The mask is auto-adjusted to extract key domain knowledge when trained on the source domain. To maintain previously learned domain knowledge, certain mask weights are frozen during adaptation, while other weights are adjusted to mitigate domain shifts with pseudo-labeled samples generated in the target domain. Our empirical results on four benchmark datasets suggest that our approach significantly enhances the performance of pretrained QA models on the target domain, and even outperforms models that have access to the source data during adaptation.
Maxwell J. Yin, Boyu Wang 0004, Yue Dong 0002, Charles Ling 0001
Trans. Assoc. Comput. Linguistics2
2024 Spectral-Spatial Attention Alignment for Multi-Source Domain Adaptation in EEG-Based Emotion Recognition
abstract
In electroencephalographic-based (EEG-based) emotion recognition, high non-stationarity and individual differences in EEG signals could lead to significant discrepancies between sessions/subjects, making generalization to a new session/subject very difficult. Most existing domain adaptation (DA) and multi-source domain adaptation (MSDA) techniques aim to mitigate this discrepancy by aligning feature distributions. However, when confronted with many diverse domain distributions, learning domain-invariant features via aligning pairwise feature distributions between domains can be hard or even counterproductive. To address this issue, this article proposes an attention alignment approach to learning abundant domain-invariant features. The motivation is simple: despite individual differences causing significant differences in feature distributions in EEG-based emotion recognition, shared affective cognitive attributes (attention) of spectral and spatial domains can be observed within the same emotion categories. The proposed spectral-spatial attention alignment multi-source domain adaptation (S2A2-MSDA) constructs domain attention to represent affective cognition attributes in spatial and spectral domains and utilizes domain consistent loss to align them between domains. Furthermore, to facilitate discriminative feature learning on the target classes, S2A2-MSDA learns the conditional semantic information of the target domain using a pseudo-labeling method. This algorithm has been validated on the SEED and SEED-IV datasets in cross-session and cross-subject scenarios, respectively. Experimental results demonstrate that S2A2-MSDA outperforms existing representative DA and MSDA methods, achieving state-of-the-art performance.
Yi Yang 0067, Ze Wang 0001, Xucheng Liu, Ziyu Jia, Boyu Wang 0004, Feng Wan 0003
IEEE Trans. Affect. Comput.6
2024 Decentralized Federated Learning: A Survey on Security and Privacy
abstract
Federated learning has been rapidly evolving and gaining popularity in recent years due to its privacy-preserving features, among other advantages. Nevertheless, the exchange of model updates and gradients in this architecture provides new attack surfaces for malicious users of the network which may jeopardize the model performance and user and data privacy. For this reason, one of the main motivations for decentralized federated learning is to eliminate server-related threats by removing the server from the network and compensating for it through technologies such as blockchain. However, this advantage comes at the cost of challenging the system with new privacy threats. Thus, performing a thorough security analysis in this new paradigm is necessary. This survey studies possible variations of threats and adversaries in decentralized federated learning and overviews the potential defense mechanisms. Trustability and verifiability of decentralized federated learning are also considered in this study.
Ehsan Hallaji, Roozbeh Razavi-Far, Mehrdad Saif, Boyu Wang 0004
IEEE Trans. Big Data4
2024 Temporal Attention Source-Free Adaptation for Chemical Processes Fault Diagnosis
abstract
Recently, domain adaptation (DA)-based fault diagnosis approaches have been actively studied in chemical processes to build a reliable fault diagnosis model for a new operating mode (i.e., target domain) by making use of labeled data from a historical mode (i.e., source domain). However, this raises privacy concerns, such as data leakage, since industrial data contains sensitive production information. Moreover, preprocessed source and target data used to train an effective target model will result in additional computational costs. Therefore, it is crucial to develop a novel privacy preserving DA-based fault diagnosis approach that can improve the diagnosis performance for a new mode and protect the privacy of a historical mode simultaneously. To this end, fault diagnosis is formulated as thesource-free DAproblem and proposes a temporal attention source-free adaptation (TASFA) algorithm, which only utilizes the pretrained source model and unlabeled target data to learn a diagnosis model. Specifically, for the time-series process, an attention mechanism is designed to capture and leverage the temporal correlations between source and target domains by extracting the most transferable information from the target time series. Empirical results on both the Tennessee Eastman process and the continuous stirred tank reactor demonstrate the effectiveness and efficiency of TASFA.
Yutang Xiao, Hongbo Shi 0002, Shuai Tan 0001, Boyu Wang 0004
IEEE Trans. Ind. Informatics6
2024 PLBR: A Semi-Supervised Document Key Information Extraction via Pseudo-Labeling Bias Rectification
abstract
Document key information extraction (DKIE) methods often require a large number of labeled samples, imposing substantial annotation costs in practical scenarios. Fortunately, pseudo-labeling based semi-supervised learning (PSSL) algorithms provide an effective paradigm to alleviate the reliance on labeled data by leveraging unlabeled data. However, the main challenges for PSSL in DKIE tasks: 1) context dependency of DKIE results in incorrect pseudo-labels. 2) high intra-class variance and low inter-class variation on DKIE. To this end, this paper proposes a similarity matrix Pseudo-Label Bias Rectification (PLBR) semi-supervised method for DKIE tasks, which improves the quality of pseudo-labels on DKIE benchmarks with rare labels. More specifically, the Similarity Matrix Bias Rectification (SMBR) module is proposed to improve the quality of pseudo-labels, which utilizes the contextual information of DKIE data through the analysis of similarity between labeled and unlabeled data. Moreover, a dual branch adaptive alignment (DBAA) mechanism is designed to adaptively align intra-class variance and alleviate inter-class variation on DKIE benchmarks, which is composed of two adaptive alignment ways. One is the intra-class alignment branch, which is designed to adaptively align intra-class variance. The other one is the inter-class alignment branch, which is developed to adaptively alleviate inter-class variance changes on the representation level. Extensive experiment results on two benchmarks demonstrate that PLBR achieves state-of-the-art performance and its performance surpasses the previous SOTA by$2.11\% \sim 2.53\%$,$2.09\% \sim 2.49\%$F1-score on FUNSD and CORD with rare labeled samples, respectively. Code will be open to the public.
Yonghong Song, Boyu Wang 0004, Jiaohao Liu, Qi Zhang 0096
IEEE Trans. Knowl. Data Eng.3
2024 Hessian Aware Low-Rank Perturbation for Order-Robust Continual Learning
abstract
Continual learning aims to learn a series of tasks sequentially without forgetting the knowledge acquired from the previous ones. In this work, we propose the Hessian Aware Low-Rank Perturbation algorithm for continual learning. By modeling the parameter transitions along the sequential tasks with the weight matrix transformation, we propose to apply the low-rank approximation on the task-adaptive parameters in each layer of the neural networks. Specifically, we theoretically demonstrate the quantitative relationship between the Hessian and the proposed low-rank approximation. The approximation ranks are then globally determined according to the marginal change of the empirical loss estimated by the layer-specific gradient and low-rank approximation error. Furthermore, we control the model capacity by pruning less important parameters to diminish the parameter growth. We conduct extensive experiments on various benchmarks, including a dataset with large-scale tasks, and compare our method against some recent state-of-the-art methods to demonstrate the effectiveness and scalability of our proposed method. Empirical results show that our method performs better on different benchmarks, especially in achieving task order robustness and handling the forgetting issue.
Jiaqi Li 0005, Yuanhao Lai, Rui Wang 0121, Changjian Shui, Sabyasachi Sahoo, Charles Ling 0001, Boyu Wang 0004, Christian Gagné 0001, Fan Zhou 0006
IEEE Trans. Knowl. Data Eng.8
2024 CROMOSim: A Deep Learning-Based Cross-Modality Inertial Measurement Simulator
abstract
With the prevalence of wearable devices, inertial measurement unit (IMU) data has been utilized in monitoring and assessing human mobility such as human activity recognition (HAR) and human pose estimation (HPE). Training deep neural network (DNN) models for these tasks require a large amount of labelled data, which are hard to acquire in uncontrolled environments. To mitigate the data scarcity problem, we design CROMOSim, a cross-modality sensor simulator that simulates high fidelity virtual IMU sensor data from motion capture systems or monocular RGB cameras. It utilizes a skinned multi-person linear model (SMPL) for 3D body pose and shape representations to enable simulation from arbitrary on-body positions. Then a DNN model is trained to learn the functional mapping from imperfect trajectory estimations in a 3D SMPL body tri-mesh due to measurement noise, calibration errors, occlusion and other modelling artifacts, to IMU data. We evaluate the fidelity of CROMOSim simulated data and its utility in data augmentation on various HAR and HPE datasets. Extensive empirical results show that the proposed model achieves a 6.7% improvement over baseline methods in a HAR task.
Yujiao Hao, Xijian Lou, Boyu Wang 0004, Rong Zheng 0001
IEEE Trans. Mob. Comput.3
2024 Automatic Metric Search for Few-Shot Learning
abstract
Few-shot learning (FSL) aims to learn a model that can identify unseen classes using only a few training samples from each class. Most of the existing FSL methods adopt a manually predefined metric function to measure the relationship between a sample and a class, which usually require tremendous efforts and domain knowledge. In contrast, we propose a novel model called automatic metric search (Auto-MS), in which an Auto-MS space is designed for automatically searching task-specific metric functions. This allows us to further develop a new searching strategy to facilitate automated FSL. More specifically, by incorporating the episode-training mechanism into the bilevel search strategy, the proposed search strategy can effectively optimize the network weights and structural parameters of the few-shot model. Extensive experiments on the miniImageNet and tieredImageNet datasets demonstrate that the proposed Auto-MS achieves superior performance in FSL problems.
Yuan Zhou 0006, Jieke Hao, Shuwei Huo, Boyu Wang 0004, Leijiao Ge, Sun-Yuan Kung
IEEE Trans. Neural Networks Learn. Syst.4
2024 Clustering Environment Aware Learning for Active Domain Adaptation
abstract
Despite the significant progress in unsupervised domain adaptation (UDA), the performance of UDA methods is still far inferior to that of the fully supervised ones. In practical scenarios, it is usually feasible to acquire labels on a small portion of the target data through active learning (AL), which aims to train an effective model with as few queried instances as possible. However, due to the domain shift, the instances selected by existing AL algorithms can be uninformative, redundant, or outlying. To address this issue, we propose a novel approach, namely, clustering environment-aware learning (CEAL), for active domain adaptation (ADA). CEAL selects potentially the most valuable instances under domain shift by exploring the informativeness and representativeness of target samples in a clustering environment-aware manner. Specifically, for the informativeness, we not only leverage the knowledge of individual points but also their nearby neighbors, by measuring the proposed clustering environment aware informativeness score (CEAIS), thus ensuring that the selected samples are highly informative. For the representativeness, we design two schemes called point distance release (PDR) and informativeness score difference exclusion (ISDE) to guarantee the diversity and validity of the selected samples. Furthermore, we fully utilize the large amount of unlabeled data from target domain via pseudo labeling and adopt information maximization to improve the reliability of the target pseudo labels, thereby further improving the performance of the model. The effectiveness of our method is empirically verified on various benchmark datasets against recent state-of-the-art algorithms.
Jian Zhu 0001, Qintai Hu, Yutang Xiao, Boyu Wang 0004, Bin Sheng 0001, C. L. Philip Chen
IEEE Trans. Syst. Man Cybern. Syst.5
2023 Class Overwhelms: Mutual Conditional Blended-Target Domain Adaptation
abstract
Current methods of blended targets domain adaptation (BTDA) usually infer or consider domain label information but underemphasize hybrid categorical feature structures of targets, which yields limited performance, especially under the label distribution shift. We demonstrate that domain labels are not directly necessary for BTDA if categorical distributions of various domains are sufficiently aligned even facing the imbalance of domains and the label distribution shift of classes. However, we observe that the cluster assumption in BTDA does not comprehensively hold. The hybrid categorical feature space hinders the modeling of categorical distributions and the generation of reliable pseudo labels for categorical alignment. To address these, we propose a categorical domain discriminator guided by uncertainty to explicitly model and directly align categorical distributions P(Z|Y). Simultaneously, we utilize the low-level features to augment the single source features with diverse target styles to rectify the biased classifier P(Y|Z) among diverse targets. Such a mutual conditional alignment of P(Z|Y) and P(Y|Z) forms a mutual reinforced mechanism. Our approach outperforms the state-of-the-art in BTDA even compared with methods utilizing domain labels, especially under the label distribution shift, and in single target DA on DomainNet.
Pengcheng Xu 0008, Boyu Wang 0004, Charles Ling 0001
AAAI2
2023 Foresee What You Will Learn: Data Augmentation for Domain Generalization in Non-stationary Environment
abstract
Existing domain generalization aims to learn a generalizable model to perform well even on unseen domains. For many real-world machine learning applications, the data distribution often shifts gradually along domain indices. For example, a self-driving car with a vision system drives from dawn to dusk, with the sky gradually darkening. Therefore, the system must be able to adapt to changes in ambient illuminations and continue to drive safely on the road. In this paper, we formulate such problems as Evolving Domain Generalization, where a model aims to generalize well on a target domain by discovering and leveraging the evolving pattern of the environment. We then propose Directional Domain Augmentation (DDA), which simulates the unseen target features by mapping source data as augmentations through a domain transformer. Specifically, we formulate DDA as a bi-level optimization problem and solve it through a novel meta-learning approach in the representation space. We evaluate the proposed method on both synthetic datasets and real-world datasets, and empirical results show that our approach can outperform other existing methods.
Qiuhao Zeng, Wei Wang 0036, Fan Zhou 0006, Charles Ling 0001, Boyu Wang 0004
AAAI5
2023 Geometric Visual Similarity Learning in 3D Medical Image Self-Supervised Pre-training
abstract
Learning inter-image similarity is crucial for 3D medical images self-supervised pre-training, due to their sharing of numerous same semantic regions. However, the lack of the semantic prior in metrics and the semantic-independent variation in 3D medical images make it challenging to get a reliable measurement for the inter-image similarity, hindering the learning of consistent representation for same semantics. We investigate the challenging problem of this task, i.e., learning a consistent representation between images for a clustering effect of same semantic features. We propose a novel visual similarity learning paradigm, Geometric Visual Similarity Learning, which embeds the prior of topological invariance into the measurement of the inter-image similarity for consistent representation of semantic regions. To drive this paradigm, we further construct a novel geometric matching head, the Z-matching head, to collaboratively learn the global and local similarity of semantic regions, guiding the efficient representation learning for different scale-level inter-image semantic features. Our experiments demonstrate that the pre-training with our learning of inter-image similarity yields more powerful inner-scene, inter-scene, and global-local transferring ability on four challenging 3D medical image tasks. Our codes and pre-trained models will be publicly available11https://github.com/YutingHe-list/GVSL.
Yuting He 0001, Guanyu Yang 0001, Rongjun Ge, Yang Chen 0008, Jean-Louis Coatrieux, Boyu Wang 0004, Shuo Li 0001
CVPR6
2023 Dynamically Instance-Guided Adaptation: A Backward-free Approach for Test-Time Domain Adaptive Semantic Segmentation
abstract
In this paper, we study the application of Test-time domain adaptation in semantic segmentation (TTDA-Seg) where both efficiency and effectiveness are crucial. Existing methods either have low efficiency (e.g., backward optimization) or ignore semantic adaptation (e.g., distribution alignment). Besides, they would suffer from the accumulated errors caused by unstable optimization and abnormal distributions. To solve these problems, we propose a novel backward-free approach for TTDA-Seg, called Dynamically Instance-Guided Adaptation (DIGA). Our principle is utilizing each instance to dynamically guide its own adaptation in a non-parametric way, which avoids the error accumulation issue and expensive optimizing cost. Specifically, DIGA is composed of a distribution adaptation module (DAM) and a semantic adaptation module (SAM), enabling us to jointly adapt the model in two indispensable aspects. DAM mixes the instance and source BN statistics to encourage the model to capture robust representation. SAM combines the historical prototypes with instance-level prototypes to adjust semantic predictions, which can be associated with the parametric classifier to mutually benefit the final results. Extensive experiments evaluated on five target domains demonstrate the effectiveness and efficiency of the proposed method. Our DIGA establishes new state-of-the-art performance in TTDA-Seg. Source code is available at: https://github.com/Waybaba/DIGA.
Zhun Zhong, Weijie Wang 0002, Charles Ling 0001, Boyu Wang 0004, Nicu Sebe
CVPR6
2023 When Source-Free Domain Adaptation Meets Learning with Noisy Labels
Gezheng Xu, Pengcheng Xu 0008, Jiaqi Li 0005, Ruizhi Pu, Charles Ling 0001, A. Ian McLeod, Boyu Wang 0004
ICLR8
2023 A Unified Solution for Privacy and Communication Efficiency in Vertical Federated Learning
abstract
Vertical Federated Learning (VFL) is a collaborative machine learning paradigm that enables multiple participants to jointly train a model on their private data without sharing it. To make VFL practical, privacy security and communication efficiency should both be satisfied. Recent research has shown that Zero-Order Optimization (ZOO) in VFL can effectively conceal the internal information of the model without adding costly privacy protective add-ons, making it a promising approach for privacy and efficiency. However, there are still two key problems that have yet to be resolved. First, the convergence rate of ZOO-based VFL is significantly slower compared to gradient-based VFL, resulting in low efficiency in model training and more communication round, which hinders its application on large neural networks. Second, although ZOO-based VFL has demonstrated resistance to state-of-the-art (SOTA) attacks, its privacy guarantee lacks a theoretical explanation. To address these challenges, we propose a novel cascaded hybrid optimization approach that employs a zeroth-order (ZO) gradient on the most critical output layer of the clients, with other parts utilizing the first-order (FO) gradient. This approach preserves the privacy protection of ZOO while significantly enhancing convergence. Moreover, we theoretically prove that applying ZOO to the VFL is equivalent to adding Gaussian Mechanism to the gradient information, which offers an implicit differential privacy guarantee. Experimental results demonstrate that our proposed framework achieves similar utility as the Gaussian mechanism under the same privacy budget, while also having significantly lower communication costs compared with SOTA communication-efficient VFL frameworks.
Ganyu Wang, Bin Gu 0001, Xiang Li 0012, Boyu Wang 0004, Charles Ling 0001
NeurIPS5
2023 Gap Minimization for Knowledge Sharing and Transfer
abstract
Learning from multiple related tasks by knowledge sharing and transfer has become increasingly relevant over the last two decades. In order to successfully transfer information from one task to another, it is critical to understand the similarities and differences between the domains. In this paper, we introduce the notion of performance gap, an intuitive and novel measure of the distance between learning tasks. Unlike existing measures which are used as tools to bound the difference of expected risks between tasks (e.g., $\mathcal{H}$-divergence or discrepancy distance), we theoretically show that the performance gap can be viewed as a data- and algorithm-dependent regularizer, which controls the model complexity and leads to finer guarantees. More importantly, it also provides new insights and motivates a novel principle for designing strategies for knowledge sharing and transfer: gap minimization. We instantiate this principle with two algorithms: 1. gapBoost, a novel and principled boosting algorithm that explicitly minimizes the performance gap between source and target domains for transfer learning; and 2. gapMTNN, a representation learning algorithm that reformulates gap minimization as semantic conditional matching for multitask learning. Our extensive evaluation on both transfer learning and multitask learning benchmark data sets shows that our methods outperform existing baselines.
Boyu Wang 0004, Jorge A. Mendez, Changjian Shui, Fan Zhou 0006, Di Wu 0044, Gezheng Xu, Christian Gagné 0001, Eric Eaton
J. Mach. Learn. Res.1
2023 Label shift conditioned hybrid querying for deep active learning
Jiaqi Li 0005, Haojia Kong, Gezheng Xu, Changjian Shui, Ruizhi Pu, Zhao Kang 0001, Charles Ling 0001, Boyu Wang 0004
Knowl. Based Syst.8
2023 Hierarchical full-attention neural architecture search based on search space compression
abstract
Neural architecture search (NAS) has significantly advanced the automatic design of convolutional neural architectures. However, it is challenging to directly extend existing NAS methods to attention networks because of the uniform structure of the search space and the lack of long-range feature extraction. To address these issues, we construct a hierarchical search space that allows various attention operations to be adopted for different layers of a network. To reduce the complexity of the search, a low-cost search space compression method is proposed to automatically remove the unpromising candidate operations for each layer. Furthermore, we propose a novel search strategy combining a self-supervised search with a supervised one to simultaneously capture long-range and short-range dependencies. To verify the effectiveness of the proposed methods, we conduct extensive experiments on various learning tasks, including image classification , fine-grained image recognition, and zero-shot image retrieval . The empirical results show strong evidence that our method is capable of discovering high-performance full-attention architectures while guaranteeing the required search efficiency.
Yuan Zhou 0006, Shuwei Huo, Boyu Wang 0004
Knowl. Based Syst.4
2023 From sMRI to task-fMRI: A unified geometric deep learning framework for cross-modal brain anatomo-functional mapping
Taicheng Huang, Zonglei Zhen, Boyu Wang 0004, Xia Wu 0001, Shuo Li 0001
Medical Image Anal.4
2023 On the value of label and semantic information in domain generalization
Fan Zhou 0006, Yuyi Chen, Boyu Wang 0004, Brahim Chaib-draa
Neural Networks4
2023 Episodic task agnostic contrastive training for multi-task learning
Fan Zhou 0006, Yuyi Chen, Jun Wen 0001, Qiuhao Zeng, Changjian Shui, Charles Ling 0001, Boyu Wang 0004
Neural Networks8
2023 Lifelong Online Learning from Accumulated Knowledge
abstract
In this article, we formulate lifelong learning as an online transfer learning procedure over consecutive tasks, where learning a given task depends on the accumulated knowledge. We propose a novel theoretical principled framework, lifelong online learning, where the learning process for each task is in an incremental manner. Specifically, our framework is composed of two-level predictions: the prediction information that is solely from the current task; and the prediction from the knowledge base by previous tasks. Moreover, this article tackled several fundamental challenges: arbitrary or even non-stationary task generation process, an unknown number of instances in each task, and constructing an efficient accumulated knowledge base. Notably, we provide a provable bound of the proposed algorithm, which offers insights on the how the accumulated knowledge improves the predictions. Finally, empirical evaluations on both synthetic and real datasets validate the effectiveness of the proposed algorithm.
Changjian Shui, William Wei Wang, Ihsen Hedhli, Chiman Wong, Feng Wan 0003, Boyu Wang 0004, Christian Gagné 0001
ACM Trans. Knowl. Discov. Data6
2023 Towards More General Loss and Setting in Unsupervised Domain Adaptation
abstract
In this article, we present an analysis of unsupervised domain adaptation with a series of theoretical and algorithmic results. We derive a novel Rényi-$\alpha$divergence-based generalization bound, which is tailored to domain adaptation algorithms with arbitrary loss functions in a stochastic setting. Moreover, our theoretical results provide new insights into the assumptions for successful domain adaptation: the closeness between the conditional distributions of the domains and the Lipschitzness on the source domain. With these assumptions, we reveal the following: if their conditional generation distributions are close, the Lipschitzness property of the target domain can be transferred from the Lipschitzness on the source domain, without knowing the exact target distribution. Motivated by our analysis and assumptions, we further derive practical principles for deep domain adaptation: 1) Rényi-2 adversarial training for marginal distributions matching and 2) Lipschitz regularization for the classifier. Our experimental results on both synthetic and real-world datasets support our theoretical findings and the practical efficiency of the proposed principles.
Changjian Shui, Ruizhi Pu, Gezheng Xu, Jun Wen 0001, Fan Zhou 0006, Christian Gagné 0001, Charles Ling 0001, Boyu Wang 0004
IEEE Trans. Knowl. Data Eng.8
2023 On the Benefits of Two Dimensional Metric Learning
abstract
In this paper, we study two dimensional metric learning (2DML) for matrix data from both theoretical and algorithmic perspectives. We first investigate the generalization bounds of 2DML based on the notion of Rademacher complexity, which theoretically justifies the benefits of learning from matrices directly. Furthermore, we present a novel boosting-based algorithm that scales well with the feature dimension. Finally, we introduce an efficient rank-one correction algorithm, which is tailored to our boosting learning procedure to produce a low-rank solution to 2DML. As our algorithm works directly on the data in matrix representation, it scales well with the feature dimension, keeps the structure and dependence in the data, and has a more compact structure and much fewer parameters to optimize. Extensive evaluations on several benchmark data sets also empirically verify the effectiveness and efficiency of our algorithm.
Di Wu 0044, Fan Zhou 0006, Boyu Wang 0004, Qicheng Lao, Chiman Wong, Changjian Shui, Yuan Zhou 0006, Feng Wan 0003
IEEE Trans. Knowl. Data Eng.3
2022 On Learning Contrastive Representations for Learning with Noisy Labels
abstract
Deep neural networks are able to memorize noisy labels easily with a softmax cross entropy (CE) loss. Previous studies attempted to address this issue focus on incorporating a noise-robust loss function to the CE loss. However, the memorization issue is alleviated but still remains due to the non-robust CE loss. To address this issue, we focus on learning robust contrastive representations of data on which the classifier is hard to memorize the label noise under the CE loss. We propose a novel contrastive regularization function to learn such representations over noisy data where label noise does not dominate the representation learning. By theoretically investigating the representations induced by the proposed regularization function, we reveal that the learned representations keep information related to true labels and discard information related to corrupted labels. Moreover, our theoretical results also indicate that the learned representations are robust to the label noise. The effectiveness of this method is demonstrated with experiments on benchmark datasets.
Qi She, A. Ian McLeod, Boyu Wang 0004
CVPR5
2022 Weakly Supervised Object Localization as Domain Adaption
abstract
Weakly supervised object localization (WSOL) focuses on localizing objects only with the supervision of image-level classification masks. Most previous WSOL methods follow the classification activation map (CAM) that localizes objects based on the classification structure with the multi-instance learning (MIL) mechanism. However, the MIL mechanism makes CAM only activate discriminative object parts rather than the whole object, weakening its performance for localizing objects. To avoid this problem, this work provides a novel perspective that models WSOL as a domain adaption (DA) task, where the score estimator trained on the source/image domain is tested on the target/pixel domain to locate objects. Under this perspective, a DA-WSOL pipeline is designed to better engage DA approaches into WSOL to enhance localization performance. It utilizes a proposed target sampling strategy to select different types of target samples. Based on these types of target samples, domain adaption localization (DAL) loss is elaborated. It aligns the feature distribution between the two domains by DA and makes the estimator perceive target domain cues by Universum regularization. Experiments show that our pipeline outperforms SOTA methods on multi benchmarks. Code are released at https://github.com/zh460045050/DA-WSOL_CVPR2022.
Lei Zhu 0012, Qi She, Yunfei You, Boyu Wang 0004, Yanye Lu
CVPR5
2022 Fair Representation Learning through Implicit Path Alignment
abstract
We consider a fair representation learning perspective, where optimal predictors, on top of the data representation, are ensured to be invariant with respect to different sub-groups. Specifically, we formulate this intuition as a bi-level optimization, where the representation is learned in the outer-loop, and invariant optimal group predictors are updated in the inner-loop. Moreover, the proposed bi-level objective is demonstrated to fulfill the sufficiency rule, which is desirable in various practical scenarios but was not commonly studied in the fair learning. Besides, to avoid the high computational and memory cost of differentiating in the inner-loop of bi-level objective, we propose an implicit path alignment algorithm, which only relies on the solution of inner optimization and the implicit differentiation rather than the exact optimization path. We further analyze the error gap of the implicit approach and empirically validate the proposed method in both classification and regression settings. Experimental results show the consistently better trade-off in prediction performance and fairness measurement.
Changjian Shui, Qi Chen 0015, Jiaqi Li 0005, Boyu Wang 0004, Christian Gagné 0001
ICML4
2022 On Learning Fairness and Accuracy on Multiple Subgroups
abstract
We propose an analysis in fair learning that preserves the utility of the data while reducing prediction disparities under the criteria of group sufficiency. We focus on the scenario where the data contains multiple or even many subgroups, each with limited number of samples. As a result, we present a principled method for learning a fair predictor for all subgroups via formulating it as a bilevel objective. Specifically, the subgroup specific predictors are learned in the lower-level through a small amount of data and the fair predictor. In the upper-level, the fair predictor is updated to be close to all subgroup specific predictors. We further prove that such a bilevel objective can effectively control the group sufficiency and generalization error. We evaluate the proposed framework on real-world datasets. Empirical evidence suggests the consistently improved fair predictions, as well as the comparable accuracy to the baselines.
Changjian Shui, Gezheng Xu, Qi Chen 0015, Jiaqi Li 0005, Charles Ling 0001, Tal Arbel, Boyu Wang 0004, Christian Gagné 0001
NeurIPS7
2022 A novel domain adaptation theory with Jensen-Shannon divergence
Changjian Shui, Qi Chen 0015, Jun Wen 0001, Fan Zhou 0006, Christian Gagné 0001, Boyu Wang 0004
Knowl. Based Syst.6
2022 On the benefits of representation regularization in invariance based domain generalization
abstract
A crucial aspect of reliable machine learning is to design a deployable system for generalizing new related but unobserved environments. Domain generalization aims to alleviate such a prediction gap between the observed and unseen environments. Previous approaches commonly incorporated learning the invariant representation for achieving good empirical performance. In this paper, we reveal that merely learning the invariant representation is vulnerable to the related unseen environment. To this end, we derive a novel theoretical analysis to control the unseen test environment error in the representation learning, which highlights the importance of controlling the smoothness of representation. In practice, our analysis further inspires an efficient regularization method to improve the robustness in domain generalization. The proposed regularization is orthogonal to and can be straightforwardly adopted in existing domain generalization algorithms that ensure invariant representation learning. Empirical results show that our algorithm outperforms the base versions in various datasets and invariance criteria.
Changjian Shui, Boyu Wang 0004, Christian Gagné 0001
Mach. Learn.2
2022 Invariant Feature Learning for Sensor-Based Human Activity Recognition
abstract
Wearable sensor-based human activity recognition (HAR) has been a research focus in the field of ubiquitous and mobile computing for years. In recent years, many deep models have been applied to HAR problems. However, deep learning methods typically require a large amount of data for models to generalize well. Significant variances caused by different participants or diverse sensor devices limit the direct application of a pre-trained model to a subject or device that has not been seen before. To address these problems, we present an invariant feature learning framework (IFLF) that extracts common information shared across subjects and devices. IFLF incorporates two learning paradigms: 1) meta-learning to capture robust features across seen domains and adapt to an unseen one with similarity-based data selection; 2) multi-task learning to deal with data shortage and enhance overall performance via knowledge sharing among different subjects. Experiments demonstrated that IFLF is effective in handling both subject and device diversion across popular open datasets and an in-house dataset. It outperforms a baseline model of up to 40 percent in test accuracy.
Yujiao Hao, Rong Zheng 0001, Boyu Wang 0004
IEEE Trans. Mob. Comput.3
2021 Multi-task Learning by Leveraging the Semantic Information
abstract
One crucial objective of multi-task learning is to align distributions across tasks so that the information between them can be transferred and shared. However, existing approaches only focused on matching the marginal feature distribution while ignoring the semantic information, which may hinder the learning performance. To address this issue, we propose to leverage the label information in multi-task learning by exploring the semantic conditional relations among tasks. We first theoretically analyze the generalization bound of multi-task learning based on the notion of Jensen-Shannon divergence, which provides new insights into the value of label information in multi-task learning. Our analysis also leads to a concrete algorithm that jointly matches the semantic distribution and controls label distribution divergence. To confirm the effectiveness of the proposed method, we first compare the algorithm with several baselines on some benchmarks and then test the algorithms under label space shift conditions. Empirical results demonstrate that the proposed method could outperform most baselines and achieve state-of-the-art performance, particularly showing the benefits under the label shift conditions.
Fan Zhou 0006, Brahim Chaib-draa, Boyu Wang 0004
AAAI3
2021 Aggregating From Multiple Target-Shifted Sources
abstract
Multi-source domain adaptation aims at leveraging the knowledge from multiple tasks for predicting a related target domain. Hence, a crucial aspect is to properly combine different sources based on their relations. In this paper, we analyzed the problem for aggregating source domains with different label distributions, where most recent source selection approaches fail. Our proposed algorithm differs from previous approaches in two key ways: the model aggregates multiple sources mainly through the similarity of semantic conditional distribution rather than marginal distribution; the model proposes a unified framework to select relevant sources for three popular scenarios, i.e., domain adaptation with limited label on target domain, unsupervised domain adaptation and label partial unsupervised domain adaption. We evaluate the proposed method through extensive experiments. The empirical results significantly outperform the baselines.
Changjian Shui, Zijian Li 0001, Jiaqi Li 0005, Christian Gagné 0001, Charles Ling 0001, Boyu Wang 0004
ICML6
2021 Domain generalization via optimal transport with metric similarity learning
Fan Zhou 0006, Zhuqing Jiang, Changjian Shui, Boyu Wang 0004, Brahim Chaib-draa
Neurocomputing4
2021 Multi-view subspace clustering via partition fusion
Juncheng Lv, Zhao Kang 0001, Boyu Wang 0004, Luping Ji, Zenglin Xu
Inf. Sci.3
2021 Discriminative active learning for domain adaptation
Fan Zhou 0006, Changjian Shui, Bincheng Huang, Boyu Wang 0004, Brahim Chaib-draa
Knowl. Based Syst.5
2021 Common Spatial Pattern Reformulated for Regularizations in Brain-Computer Interfaces
abstract
Common spatial pattern (CSP) is one of the most successful feature extraction algorithms for brain-computer interfaces (BCIs). It aims to find spatial filters that maximize the projected variance ratio between the covariance matrices of the multichannel electroencephalography (EEG) signals corresponding to two mental tasks, which can be formulated as a generalized eigenvalue problem (GEP). However, it is challenging in principle to impose additional regularization onto the CSP to obtain structural solutions (e.g., sparse CSP) due to the intrinsic nonconvexity and invariance property of GEPs. This article reformulates the CSP as a constrained minimization problem and establishes the equivalence of the reformulated and the original CSPs. An efficient algorithm is proposed to solve this optimization problem by alternately performing singular value decomposition (SVD) and least squares. Under this new formulation, various regularization techniques for linear regression can then be easily implemented to regularize the CSPs for different learning paradigms, such as the sparse CSP, the transfer CSP, and the multisubject CSP. Evaluations on three BCI competition datasets show that the regularized CSP algorithms outperform other baselines, especially for the high-dimensional small training set. The extensive results validate the efficiency and effectiveness of the proposed CSP formulation in different learning contexts.
Boyu Wang 0004, Chiman Wong, Zhao Kang 0001, Feng Liu 0011, Changjian Shui, Feng Wan 0003, C. L. Philip Chen
IEEE Trans. Cybern.1
2021 Blood Pressure States Transition Inference Based on Multi-State Markov Model
abstract
The investigation of risk factors associated with hypertension patients has been extensively studied in the past decades. However, the pattern of natural progressive trajectories to hypertension from nonhypertensive states was rarely explored. In this study, we are interested in discovering the underlying transition patterns between different blood pressure states, namely normal state, elevated state, and hypertensive state among the working population in the United States. A multi-state Markov model was built based on 88,966 clinical records from 34,719 participants we collected during the worksite preventive screening from 2012 to 2018. We first investigated the various risk factors, and we found that body mass index (BMI) is the most critical factor for developing new-onset hypertension. The transition probabilities, survival probabilities, and sojourn time of each state were derived given different levels of BMI, age groups, and gender categories. We found the underweight participants are more likely to remain in the current nonhypertensive states within 3 years, while extremely obese participants have a higher probability of developing hypertension. We discovered the distinct transition patterns among male and female participants. On average, the sojourn time in the normal state for normal-weight participants is 4.33 years for females and 2.18 years for their male counterparts. For the extremely obese participants, the average sojourn time in the normal state is 1.38 years for females and 0.71 years for males. In the end, a web-based graphical user interface (GUI) application was developed for clinicians to visualize the impact of behavioral interventions on delaying the progression of hypertension. Our analysis can provide a unique insight into hypertension research and proactive interventions.
Jingmei Yang, Feng Liu 0011, Boyu Wang 0004, Chaoyang Chen 0001, Timothy Church, Lee Dukes, Jeffrey O. Smith
IEEE J. Biomed. Health Informatics3
2021 Imputation-Based Ensemble Techniques for Class Imbalance Learning
abstract
Correct classification of rare samples is a vital data mining task and of paramount importance in many research domains. This article mainly focuses on the development of the novel class-imbalance learning techniques, which make use of oversampling methods integrated with bagging and boosting ensembles. Two novel oversampling strategies based on the single and the multiple imputation methods are proposed. The proposed techniques aim to create useful synthetic minority class samples, similar to the original minority class samples, by estimation of missing values that are already induced in the minority class samples. The re-balanced datasets are then used to train base-learners of the ensemble algorithms. In addition, the proposed techniques are compared with the commonly used class imbalance learning methods in terms of three performance metrics including AUC, F-measure, and G-mean over several synthetic binary class datasets. The empirical results show that the proposed multiple imputation-based oversampling combined with bagging significantly outperforms other competitors.
Roozbeh Razavi-Far, Maryam Farajzadeh-Zanjani, Boyu Wang 0004, Mehrdad Saif, Shiladitya Chakrabarti
IEEE Trans. Knowl. Data Eng.3
2021 Task Similarity Estimation Through Adversarial Multitask Neural Network
abstract
Multitask learning (MTL) aims at solving the related tasks simultaneously by exploiting shared knowledge to improve performance on individual tasks. Though numerous empirical results supported the notion that such shared knowledge among tasks plays an essential role in MTL, the theoretical understanding of the relationships between tasks and their impact on learning shared knowledge is still an open problem. In this work, we are developing a theoretical perspective of the benefits involved in using information similarity for MTL. To this end, we first propose an upper bound on the generalization error by implementing the Wasserstein distance as the similarity metric. This indicates the practical principles of applying the similarity information to control the generalization errors. Based on those theoretical results, we revisited the adversarial multitask neural network and proposed a new training algorithm to learn the task relation coefficients and neural network parameters automatically. The computer vision benchmarks reveal the abilities of the proposed algorithms to improve the empirical performance. Finally, we test the proposed approach on real medical data sets, showing its advantage for extracting task relations.
Fan Zhou 0006, Changjian Shui, Mahdieh Abbasi, Louis-Émile Robitaille, Boyu Wang 0004, Christian Gagné 0001
IEEE Trans. Neural Networks Learn. Syst.5
2020 Deep Active Learning: Unified and Principled Method for Query and Training
abstract
In this paper, we are proposing a unified and principled method for both the querying and training processes in deep batch active learning. We are providing theoretical insights from the intuition of modeling the interactive procedure in active learning as distribution matching, by adopting the Wasserstein distance. As a consequence, we derived a new training loss from the theoretical analysis, which is decomposed into optimizing deep neural network parameters and batch query selection through alternative optimization. In addition, the loss for training a deep neural network is naturally formulated as a min-max optimization problem through leveraging the unlabeled data information. Moreover, the proposed principles also indicate an explicit uncertainty-diversity trade-off in the query batch selection. Finally, we evaluate our proposed method on different benchmarks, consistently showing better empirical performances and a better time-efficient query strategy compared to the baselines.
Changjian Shui, Fan Zhou 0006, Christian Gagné 0001, Boyu Wang 0004
AISTATS4
2019 A Principled Approach for Learning Task Similarity in Multitask Learning
abstract
Multitask learning aims at solving a set of related tasks simultaneously, by exploiting the shared knowledge for improving the performance on individual tasks. Hence, an important aspect of multitask learning is to understand the similarities within a set of tasks. Previous works have incorporated this similarity information explicitly (e.g., weighted loss for each task) or implicitly (e.g., adversarial loss for feature adaptation), for achieving good empirical performances. However, the theoretical motivations for adding task similarity knowledge are often missing or incomplete. In this paper, we give a different perspective from a theoretical point of view to understand this practice. We first provide an upper bound on the generalization error of multitask learning, showing the benefit of explicit and implicit task similarity knowledge. We systematically derive the bounds based on two distinct task similarity metrics: H divergence and Wasserstein distance. From these theoretical results, we revisit the Adversarial Multi-task Neural Network, proposing a new training algorithm to learn the task relation coefficients and neural network parameters iteratively. We assess our new algorithm empirically on several benchmarks, showing not only that we find interesting and robust task relations, but that the proposed approach outperforms the baselines, reaffirming the benefits of theoretical insight in algorithm design.
Changjian Shui, Mahdieh Abbasi, Louis-Émile Robitaille, Boyu Wang 0004, Christian Gagné 0001
IJCAI4
2019 Clustering with similarity preserving
Zhao Kang 0001, Boyu Wang 0004, Hongyuan Zhu 0002, Zenglin Xu
Neurocomputing3
2018 Leveraging Disease Progression Learning for Medical Image Recognition
Qicheng Lao, Thomas Fevens, Boyu Wang 0004
BIBM3
2018 Cross-Layer Design for Network Lifetime Maximization in Underwater Wireless Sensor Networks
abstract
This paper investigates the cross-layer design problem with the goal of maximizing the network lifetime for energy-constrained underwater wireless sensor networks (UWSNs). We first jointly consider link scheduling, transmission power and transmission rate in a proposed optimization problem with the adoption of time division multiple access (TDMA) schedules. Then, we propose an iterative algorithm to solve the optimization problem. It alternates between (1) link scheduling and (2) computation of transmission powers and transmission rates. In fact, the convergence of such iterative algorithm can be mathematically and empirically supported. We evaluate our algorithm for several network topologies. Extensive simulation results demonstrate the superiority of the proposed approach.
Yuan Zhou 0006, Yu Hen Hu, Boyu Wang 0004, Sun-Yuan Kung
ICC4
2017 Boosting Based Multiple Kernel Learning and Transfer Regression for Electricity Load Forecasting
Di Wu 0044, Boyu Wang 0004, Doina Precup, Benoit Boulet
ECML/PKDD (3)2
2011 Entropy penalized learning for Gaussian mixture models
abstract
In this paper, we propose an entropy penalized approach to address the problem of learning the parameters of Gaussian mixture models (GMMs) with components of small weights. In addition, since the method is based on minimum message length (MML) criterion, it can also determine the number of components of the mixture model. The simulation results demonstrate that our method outperform several other state-of-art model selection algorithms especially for the mixtures with components of very different weights.
Boyu Wang 0004, Feng Wan 0003, Peng Un Mak, Pui-In Mak, Mang I Vai
IJCNN1
2011 A Solution to harmonic frequency problem: Frequency and phase coding-based brain-computer interface
abstract
In this paper, we propose a modified visual stimulus generation method and feature detection algorithm to design a frequency and phase coding steady-state visual evoked potential (SSVEP) based brain-computer interface (BCI). By utilizing both frequency and phase information, we solve the harmonic frequency problem in our proposed SSVEP-BCI system. The offline experimental results show that the proposed feature detection algorithm can enhance the classification rate over 10% (from 69%±12% to 82%±8%) even though only one signal electrode is used and the harmonic frequencies (6.67Hz, 13.33Hz, 8.57Hz and 17.14Hz) are employed.
Chiman Wong, Boyu Wang 0004, Feng Wan 0003, Peng Un Mak, Pui-In Mak, Mang I Vai
IJCNN2
2009 Classification of Single-Trial EEG Based on Support Vector Clustering during Finger Movement
Boyu Wang 0004, Feng Wan 0003
ISNN (2)1
2008 A modified counter-propagation network for process mean shift identification
abstract
In a control chart, unnatural patterns are always associated with some specific assignable causes that should be eliminated. The identification of control chart pattern (CCP) is therefore important and further estimation of the unnatural pattern parameters can improve the manufacturing process. In this paper, a modified counter-propagation network (m-CPN) is developed to classify the mean shift and simultaneously estimate the shift magnitude. The m-CPN is compared with five existing networks through numerical simulation and the result shows a better performance of the m-CPN in terms of classification accuracy, as well as both Type I and Type II errors.
Boyu Wang 0004, Feng Wan 0003, Lianjie Shu
SMC1