Di Wang 0015

dblp:18/5410-15 · DBLP profile ↗
← Back
16ranked-venue papers in the field
1as first author
15since 2021 · last 2026
0000-0003-4908-0243ORCID · conflict

Domains — venue-derived; a paper can count in several

Database Systems & Data Management · 8Data Mining & Knowledge Discovery · 4 (1 first)Information Retrieval & Web Search · 4
YearPublicationVenuePosition
2026 PIXEL: Adaptive Steering Via Position-wise Injection with eXact Estimated Levels under a Subspace Calibration
abstract
Reliable behavior control is central to deploying Large Language Models (LLMs) on the web. Activation steering offers a tuning-free route to align attributes (e.g., truthfulness) that ensure trustworthy generation. Prevailing approaches rely on coarse heuristics and lack a principled account of where to steer and how strongly to intervene. To this end, we propose Position-wise Injection with eXact Estimated Levels (PIXEL), a position-wise activation steering framework that, in contrast to prior work, learns a property-aligned subspace from dual views (tail-averaged and end-token) and selects intervention strength via a constrained geometric objective with a closed-form solution, thereby adapting to token-level sensitivity without global hyperparameter tuning. PIXEL further performs sample-level orthogonal residual calibration to refine the global attribute direction and employs a lightweight position-scanning routine to identify receptive injection sites. We additionally provide representation-level guarantees for the minimal-intervention rule, supporting reliable alignment. Across diverse models and evaluation paradigms, PIXEL consistently improves attribute alignment while preserving model general capabilities, offering a practical and principled method for LLMs' controllable generation. Our code is available at https://anonymous.4open.science/r/PIXEL-Adaptive-Steering-95DC
Manjiang Yu, Hongji Li 0003, Priyanka Singh 0001, Xue Li 0001, Di Wang 0015, Lijie Hu
WWW5
2026 PRISM: Link Prediction in Attributed Networks With Uncertain Modalities
abstract
Link prediction for attributed graphs has garnered significant attention due to its ability to enhance predictive performance by leveraging multi-modal node attributes. However, real-world challenges such as privacy concerns, content restrictions, and attribute constraints often result in nodes facing varying degrees of missing modalities in their attributes, significantly limiting the effectiveness of existing approaches. Building on this fact, we propose a model for linkPRediction in attrIbuted networkSwith uncertainModalities (PRISM), which learns the shared representations across various scenarios of missing modalities through dual-level adversarial training.PRISMcomprises four modules,i.e.,a GCN extractor, an adversarial extractor, an attentive fusion, and an adaptive aggregator. The GCN extractor leverages graph convolutional networks (GCN) to extract fundamental representations from the network topology. The adversarial extractor employs dual-level adversarial training to acquire the shared representations across various multi-modal scenarios at the node-level and link-level, respectively. The attentive fusion applies the multi-head attention mechanism to integrate the shared representations and the fundamental representations. The adaptive aggregator comprehensively considers both node-level and link-level representations to predict the existence of links. Experimental evaluation using real-world datasets demonstrates thatPRISMsignificantly outperforms existing state-of-the-art link prediction methods for multi-modal attributed graphs under missing modalities by improving the Recall@50 metric (R@50) by up to 38.79%.
Muhammad Asif Ali, Huan Wang 0005, Zhongfei Zhang, Junyang Chen 0001, Di Wang 0015
IEEE Trans. Knowl. Data Eng.6
2026 Generative Regularities in Multi-Layer Networks: A Shared-Latent Space Representation Approach
abstract
Understanding structural regularities across layers in multi-layer networks is essential for uncovering their underlying generative mechanisms. While link prediction has been widely explored in multi-layer networks, it is typically treated as an isolated technical problem, often missing its broader implications for network structure and the mechanisms driving edge formation. In this article, we investigate the extent to which network layers exhibit shared generative regularities. By examining the alignment of latent representations across layers, we assess the similarity of their underlying mechanisms and leverage this alignment to improve predictive performance. To facilitate this, we introduce a new metric, C ross- L ayer G enerative C onsistency ( CLGC ), which quantitatively captures the degree of structural and generative alignment between network layers. CLGC is grounded in the shared-latent space framework, positing that layers generated by similar mechanisms will produce compatible latent representations. To realize this approach, we present SupportNet – Support prediction and consistency analysis in multi-layer Net works–a GCN-based model augmented with adversarial training to effectively learn robust shared-latent space representations. These representations support both accurate link prediction and interpretable evaluation of cross-layer generative consistency. Experiments on real-world multi-layer networks demonstrate that SupportNet delivers strong link prediction results improving AUC by 17.47%, AP by 40.41% and AUPR by 39.59% on the Kapferer dataset, while CLGC reveals significant patterns of structural and generative alignment among layers.
Muhammad Asif Ali, Anyu Xue, Huan Wang 0005, Junyang Chen 0001, Di Wang 0015
ACM Trans. Web6
2025 Stable Vision Concept Transformers for Medical Diagnosis
Lijie Hu, Songning Lai, Yuan Hua, Shu Yang 0010, Jingfeng Zhang, Di Wang 0015
ECML/PKDD (3)6
2025 Differentially Private Sparse Linear Regression with Heavy-Tailed Responses
Xizhi Tian, Touming Tao, Zihang Xiang, Di Wang 0015
ECML/PKDD (5)5
2025 LUSTER: Link Prediction Utilizing Shared-Latent Space Representation in Multi-Layer Networks
abstract
Link prediction in multi-layer networks is a longstanding issue that predicts missing links based on the observed structures across all layers. Existing link prediction methods in multi-layer network typically merge the multi-layer network into a single-layer network and/or perform explicit calculations using intra-layer and inter-layer similarity metrics. However, these approaches often overlook the role of coupling in multi-layer networks, specifically the shared information and latent relationships between layers, which in turn limits prediction performance. This calls the need for methods that can extract representations in a shared-latent space to enhance inter-layer information sharing and prediction performance. In this paper, we propose a novel end-to-end framework namely: Link prediction Utilizing Shared-laTent spacE Representation (LUSTER) in multi-layer networks. LUSTER consists of four key modules: the representation extractor, the latent space learner, the complementary enhancer, and the link predictor. The representation extractor focuses on learning the intra-layer representations of each layer, capturing the data characteristics within the layer. The latent space learner extracts representations from the shared-latent space across different network layers through adversarial training. The complementary enhancer combines the intra-layer representations and the shared-latent space representations through orthogonal fusion, providing comprehensive information. Finally, the link predictor uses the enhanced representations to predict missing links. Extensive experimental analyses demonstrate that LUSTER outperforms state-of-the-art methods for link prediction in multi-layer networks, improving the AUC metric by up to 15.87%.
Muhammad Asif Ali, Huan Wang 0005, Junyang Chen 0001, Di Wang 0015
WWW5
2025 TAAD: Time-varying adversarial anomaly detection in dynamic graphs
Guanghua Liu, Jia Zhang 0022, Huan Wang 0005, Di Wang 0015
Inf. Process. Manag.6
2025 Towards Stable and Explainable Attention Mechanisms
abstract
Currently, attention mechanism has become a standard fixture in most state-of-the-art natural language processing (NLP) models, not only due to the outstanding performance it could gain but also due to plausible innate explanations for the behaviors of neural architectures it provides, which is notoriously difficult to analyze. However, recent studies show that attention is unstable against randomness and perturbations during training or testing, such as random seeds and slight perturbation of embedding vectors, which impedes it from becoming a faithful explanation tool. Thus, a natural question is whether we can find some substitute for the current attention that is more stable and could keep the most important characteristics of explanation and prediction of attention. In this paper, to resolve the problem, we provide a rigorous definition of such alternate namely SEAT (Stable andExplainableAttention). Specifically, a SEAT should have the following three properties: (1) Its prediction distribution is enforced to be close to the distribution based on the vanilla attention; (2) Its top-$k$indices have large overlaps with those of the vanilla attention; (3) It is robust w.r.t perturbations, i.e., any slight perturbation on SEAT will not change the prediction distribution too much, which implicitly indicates that it is stable to randomness and perturbations. To further improve the interpretability stability against perturbations, based on SEAT we provide another definition called SEAT++. Then we propose a method to get a SEAT++, which could be considered an ad hoc modification for canonical attention. Finally, through intensive experiments on various datasets, we compare our SEAT and SEAT++ with other baseline methods using RNN, BiLSTM, and BERT architectures via six different evaluation metrics for model interpretation, stability, and accuracy. Results show that SEAT and SEAT++ are more stable against different perturbations and randomness while also keeping the explainability of attention, which indicates they provide more faithful explanations. Moreover, compared with vanilla attention, there is almost no utility (accuracy) degradation for SEAT and SEAT++.
Lijie Hu, Yixin Liu 0002, Ninghao Liu 0001, Mengdi Huai, Lichao Sun 0001, Di Wang 0015
IEEE Trans. Knowl. Data Eng.7
2025 EPM: Evolutionary Perception Method for Anomaly Detection in Noisy Dynamic Graphs
abstract
With the rapid expansion of interactions across various domains such as knowledge graphs and social networks, anomaly detection in dynamic graphs has become increasingly critical for mitigating potential risks. However, existing anomaly detection methods often assume noise-free dynamic graphs, overlooking the prevalence of noisy dynamic graphs in real-world applications. Specifically, noisy dynamic graphs affected by structural noises-such as spurious and missing nodes and edges-struggle to consistently provide reliable structural evidence for anomaly detection. To tackle this challenge, we propose an Evolutionary Perception Method (EPM) for identifying anomalous nodes in noisy dynamic graphs by resisting the interference of structural noises. EPM primarily consists of two components: a dynamic fitter and a filtering reviser. The dynamic fitter characterizes the interaction dynamics of nodes that removes and generates links at each period as a multiple superposition state, utilizing various link prediction algorithms to fit evolutionary mechanisms. Additionally, the filtering reviser designs evolutional entropies to quantify the evolutional uncertainty in multiple superposition states, further designing the Kalman filter to optimize these entropies. Extensive experiments show that the proposed EPM method surpasses state-of-the-art approaches in detecting anomalous nodes in noisy dynamic graphs.
Huan Wang 0005, Junyang Chen 0001, Yirui Wu, Victor C. M. Leung, Di Wang 0015
IEEE Trans. Knowl. Data Eng.5
2024 Communication Efficient and Provable Federated Unlearning
abstract
We study federated unlearning, a novel problem to eliminate the impact of specific clients or data points on the global model learned via federated learning (FL). This problem is driven by the right to be forgotten and the privacy challenges in FL. We introduce a new framework for exact federated unlearning that meets two essential criteria:communication efficiencyandexact unlearning provability.To our knowledge, this is the first work to tackle both aspects coherently. We start by giving a rigorous definition ofexactfederated unlearning, which guarantees that the unlearned model is statistically indistinguishable from the one trained without the deleted data. We then pinpoint the key property that enables fast exact federated unlearning: total variation (TV) stability, which measures the sensitivity of the model parameters to slight changes in the dataset. Leveraging this insight, we develop a TV-stable FL algorithm called FATS, which modifies the classical FedAvg algorithm for TV Stability and employs local SGD with periodic averaging to lower the communication round. We also design efficient unlearning algorithms for FATS under two settings: client-level and sample-level unlearning. We provide theoretical guarantees for our learning and unlearning algorithms, proving that they achieve exact federated unlearning with reasonable convergence rates for both the original and unlearned models. We empirically validate our framework on 6 benchmark datasets, and show its superiority over state-of-the-art methods in terms of accuracy, communication cost, computation cost, and unlearning efficacy.
Youming Tao 0001, Cheng-Long Wang 0003, Miao Pan, Dongxiao Yu, Xiuzhen Cheng, Di Wang 0015
Proc. VLDB Endow.6
2024 Privacy Amplification via Shuffling: Unified, Simplified, and Tightened
abstract
The shuffle model of differential privacy provides promising privacy-utility balances in decentralized, privacy-preserving data analysis. However, the current analyses of privacy amplification via shuffling lack both tightness and generality. To address this issue, we propose the variation-ratio reduction as a comprehensive framework for privacy amplification in both single-message and multi-message shuffle protocols. It leverages two new parameterizations: the total variation bounds of local messages and the probability ratio bounds of blanket messages, to determine indistinguishability levels. Our theoretical results demonstrate that our framework provides tighter bounds, especially for local randomizers with extremal probability design, where our bounds are exactly tight. Additionally, variation-ratio reduction complements parallel composition in the shuffle model, yielding enhanced privacy accounting for popular sampling-based randomizers employed in statistical queries (e.g., range queries, marginal queries, and frequent itemset mining). Empirical findings demonstrate that our numerical amplification bounds surpass existing ones, conserving up to 30% of the budget for single-message protocols, 75% for multi-message ones, and a striking 75%-95% for parallel composition. Our bounds also result in a remarkably efficient Õ ( n ) algorithm that numerically amplifies privacy in less than 10 seconds for n = 10 8 users.
Shaowei Wang 0003, Yun Peng 0002, Jin Li 0002, Zikai Wen, Shiyu Yu, Di Wang 0015, Wei Yang 0011
Proc. VLDB Endow.7
2024 Fair Single Index Model
abstract
Single-index models (SIMs) have been widely used in various applications due to their simplicity and interpretability. However, despite the potential for SIMs to result in discriminatory outcomes based on sensitive attributes like gender, race, or ethnicity, the issue of fairness has not been thoroughly examined in recent studies on the topic. This paper aims to address these fairness concerns by proposing methods for building fair SIMs. Specifically, based on the definition of equal opportunity, we first provide a fairness definition for SIM. Next, we develop a unified fair SIM model and propose an efficient method to solve the fair SIM. Theoretically, we also show that our output is consistent in fairness. Finally, we conduct comprehensive experimental studies over 7 benchmark datasets and demonstrate that our fair SIM outperforms the other 8 baseline methods.
Yidong Wang 0006, Jinhui Xu 0001, Di Wang 0015
ACM Trans. Knowl. Discov. Data4
2024 Nearly Optimal Rates of Privacy-Preserving Sparse Generalized Eigenvalue Problem
abstract
In this paper, we study the (sparse) Generalized Eigenvalue Problem (GEP), which arises in a number of modern statistical learning models, such as principal component analysis (PCA), canonical correlation analysis (CCA), Fisher's discriminant analysis (FDA) and sliced inverse regression (SIR). We provide the first study on GEP in the differential privacy (DP) model under both deterministic and stochastic settings. In the low dimensional case, we provide a$\rho$-Concentrated DP (CDP) method namely DP-Rayleigh Flow and show if the initial vector is close enough to the optimal vector, its output has an$\ell _{2}$-norm estimation error of$\tilde{O}(\frac{d}{n}+\frac{d}{n^{2}\rho })$(under some mild assumptions), where$d$is the dimension and$n$is the sample size. Next, we discuss how to find such an initial parameter privately. In the high dimensional sparse case where$d\gg n$, we propose the DP-Truncated Rayleigh Flow method whose output could achieve an error of$\tilde{O}(\frac{s\log d}{n}+\frac{s\log d}{n^{2}\rho })$for various statistical models, where$s$is the sparsity of the underlying parameter. Moreover, we show that these errors in the stochastic setting are optimal up to a factor of$\text{Poly}(\log n)$by providing the lower bounds of PCA and SIR under the statistical setting and in the CDP model. Finally, to give a separation between$\epsilon$-DP and$\rho$-CDP for GEP, we also provide the lower bound$\Omega (\frac{d}{n}+\frac{d^{2}}{n^{2}\epsilon ^{2}})$and$\Omega (\frac{s\log d}{n}+\frac{s^{2}\log ^{2} d}{n^{2}\epsilon ^{2}})$of private minimax risk for PCA, under the statistical setting and$\epsilon$-DP model, in low and high dimensional sparse case respectively. Finally, extensive experiments on both synthetic and real-world data support our previous theoretical analysis.
Lijie Hu, Zihang Xiang, Di Wang 0015
IEEE Trans. Knowl. Data Eng.4
2023 Practical Differentially Private and Byzantine-resilient Federated Learning
abstract
Privacy and Byzantine resilience are two indispensable requirements for a federated learning (FL) system. Although there have been extensive studies on privacy and Byzantine security in their own track, solutions that consider both remain sparse. This is due to difficulties in reconciling privacy-preserving and Byzantine-resilient algorithms. In this work, we propose a solution to such a two-fold issue. We use our version of differentially private stochastic gradient descent (DP-SGD) algorithm to preserve privacy and then apply our Byzantine-resilient algorithms. We note that while existing works follow this general approach, an in-depth analysis on the interplay between DP and Byzantine resilience has been ignored, leading to unsatisfactory performance. Specifically, for the random noise introduced by DP, previous works strive to reduce its seemingly detrimental impact on the Byzantine aggregation. In contrast, we leverage the random noise to construct a first-stage aggregation that effectively rejects many existing Byzantine attacks. Moreover, based on another property of our DP variant, we form a second-stage aggregation which provides a final sound filtering. Our protocol follows the principle of co-designing both DP and Byzantine resilience. We provide both theoretical proof and empirical experiments to show our protocol is effective: retaining high accuracy while preserving the DP guarantee and Byzantine resilience. Compared with the previous work, our protocol 1) achieves significantly higher accuracy even in a high privacy regime; 2) works well even when up to 90% distributive workers are Byzantine.
Zihang Xiang, Tianhao Wang 0001, Wanyu Lin, Di Wang 0015
Proc. ACM Manag. Data4
2022 High Dimensional Differentially Private Stochastic Optimization with Heavy-tailed Data
abstract
As one of the most fundamental problems in machine learning, statistics and differential privacy, Differentially Private Stochastic Convex Optimization (DP-SCO) has been extensively studied in recent years. However, most of the previous work can only handle either regular data distributions or irregular data in the low dimensional space case. To better understand the challenges arising from irregular data distributions, in this paper we provide the first study on the problem of DP-SCO with heavy-tailed data in the high dimensional space. In the first part we focus on the problem over some polytope constraint (such as the l1-norm ball). We show that if the loss function is smooth and its gradient has bounded second order moment, it is possible to get a (high probability) error bound (excess population risk) of Õ(log d/(nε)1/3) in the ε-DP model, where n is the sample size and d is the dimension of the underlying space. Next, for LASSO, if the data distribution has bounded fourth-order moments, we improve the bound to Õ(log d/(nε)2/5) in the $(ε, δ)-DP model. In the second part of the paper, we study sparse learning with heavy-tailed data. We first revisit the sparse linear model and propose a truncated DP-IHT method whose output could achieve an error of Õ ((s*2 log2d)/nε), where s* is the sparsity of the underlying parameter. Then we study a more general problem over the sparsity (i.e., l0-norm) constraint, and show that it is possible to achieve an error of Õ((s*3/2 log d)/nε), which is also near optimal up to a factor of Õ(√s*), if the loss function is smooth and strongly convex.
Lijie Hu, Shuo Ni, Hanshen Xiao, Di Wang 0015
PODS4
2020 Escaping Saddle Points of Empirical Risk Privately and Scalably via DP-Trust Region Method
Di Wang 0015, Jinhui Xu 0001
ECML/PKDD (3)1