EDBT 2026 Demo / reviewers in the wild / expert
Kai Chen 0005
dblp:c/KaiChen5
· DBLP profile ↗
26ranked-venue papers in the field
0as first author
25since 2021 · last 2026
0000-0003-2587-6028ORCID · conflict
Domains — venue-derived; a paper can count in several
Data Mining & Knowledge Discovery · 14Database Systems & Data Management · 8Information Retrieval & Web Search · 4
| Year | Publication | Venue | Position |
|---|---|---|---|
| 2026 | Tracing paths, pruning noise: Toward robust IP geolocation via topology-guided shaping and refinement
Xueting Liu 0005, Wenxin Tai, Ting Zhong, Yong Wang 0046, Kai Chen 0005, Fan Zhou 0002 |
Inf. Process. Manag. | 5 |
| 2026 | Invariant learning improves out-of-distribution generalization for IP geolocation
Xueting Liu 0005, Wenxin Tai, Joojo Walker, Yong Wang 0046, Kai Chen 0005, Fan Zhou 0002 |
Inf. Process. Manag. | 6 |
| 2026 | GRPCI: Harnessing Temporal-Spatial Dynamics for Graph Representation LearningabstractTemporal interactions form the crux of numerous real-world scenarios, thus necessitating effective modeling in temporal graph representation learning. Despite extensive research within this domain, we identify a significant oversight in current methodologies: the temporal-spatial dynamics in graphs, encompassing both structural and temporal coherence, remain largely unaddressed. In an effort to bridge this research gap, we present a novel framework termed Graph Representation learning enhanced by Periodic and Community Interactions (GRPCI). GRPCI consists of two primary mechanisms devised explicitly to tackle the aforementioned challenge. Firstly, to utilize latent temporal dynamics, we propose a novel periodicity-based neighborhood aggregation mechanism that underscores neighbors engaged in a periodic interaction pattern. This mechanism seamlessly integrates the element of periodicity into the model. Secondly, to exploit structural dynamics, we design a novel contrastive-based local community representation learning mechanism. This mechanism features a heuristic dynamic contrastive pair sampling strategy aimed at enhancing the modeling of the latent distribution of local com munities within the graphs. Through the incorporation of these two mechanisms, GRPCI markedly augments the performance of graph networks. Empirical evaluations, conducted via a temporal link prediction task across five real-life datasets, attest to the superior performance of GRPCI in comparison to existing state of-the-art methodologies. The results of this study validate the efficacy of GRPCI, thereby establishing a new benchmark for future research in the field of temporal graph representation learning. Our findings underscore the importance of considering both temporal and structural consistency in temporal graph learning, and advocate for further exploration of this paradigm. Rong-Hua Li 0001, Zhaoxin Fan, Kai Chen 0005, Yujin Gao, Hongchao Qin, Guoren Wang |
IEEE Trans. Knowl. Data Eng. | 4 |
| 2025 | Exploiting Student Parallelism for Low-latency GPU Inference of BERT-like Models in Online ServicesabstractBERT-like models have been widely adopted in text mining and web search due to their high accuracy. However, large BERT-like models suffer from inefficient online inference on GPUs for two main reasons. First, their high accuracy relies on large model depth, which linearly increases sequential computation on GPUs. Second, stochastic and dynamic online workloads lead to extra costs due to batching and padding. To address the problem, we present Student Parallelism for efficient GPU inference of BERT-like models under real-world online workloads. At its core, Student Parallelism adopts stacking distillation and boosting ensemble, distilling the original deep model into a group of shallow but virtually stacked student models running in parallel. This enables Student Parallelism to achieve a low model depth (e.g., two layers), and thus low inference latency while maintaining accuracy. In addition, we design adaptive student pruning to adjust the number of students according to the dynamic online workloads. For example, during workload bursts, it can temporarily decrease the number of students with minimal accuracy loss to improve system throughput. Extensive experiments on real-world datasets and workloads show that Student Parallelism achieves up to 4.1× lower latency while maintaining accuracy and up to 22.27× higher throughput during workload bursts. Weiyan Wang, Yilun Jin, Yiming Zhang 0003, Victor Junqiu Wei, Han Tian, Li Chen 0008, Jinbao Xue, Yangyu Tao, Kai Chen 0005 |
KDD (2) | 10 |
| 2025 | Sequoia: An Accessible and Extensible Framework for Privacy-Preserving Machine Learning over Distributed DataabstractPrivacy-preserving machine learning (PPML) algorithms use secure computation protocols to allow multiple data parties to collaboratively train machine learning (ML) models while maintaining their data confidentiality. However, current PPML frameworks couple secure protocols with ML models in PPML algorithm implementations, making it challenging for non-experts to develop and optimize PPML applications, limiting their accessibility and performance. We propose Sequoia, a novel PPML framework that decouples ML models and secure protocols to optimize the development and execution of PPML applications across data parties. Sequoia offers JAX-compatible APIs for users to program their ML models, while using a compiler-executor architecture to automatically apply PPML algorithms and system optimizations for model execution over distributed data. The compiler in Sequoia incorporates cross-party PPML processes into user-defined ML models by transparently adding computation, encryption, and communication steps with extensible policies, and the executor efficiently schedules code execution across multiple data parties, considering data dependencies and device heterogeneity. Compared to existing PPML frameworks, Sequoia requires 64%-92% fewer lines of code for users to implement the same PPML algorithms, and achieves 88% speedup of training throughput in horizontal PPML. Kaiqiang Xu, Di Chai, Junxue Zhang 0001, Fan Lai 0001, Kai Chen 0005 |
Proc. ACM Manag. Data | 5 |
| 2024 | Triple-D: Denoising Distant Supervision for High-Quality Data CreationabstractDistant supervision is a technique that aims to create large amounts of training data at a low cost. This approach benefits various downstream systems, particularly in natural language processing and relation extraction tasks. However, due to its strong assumption that any sentence containing entities expresses the specific relation between them found in existing knowledge bases (KBs), distant supervision introduces considerable noise. Existing works attempt to denoise distant supervision data by either using the original text or replacing entities in the text with patterns representing the entity types as inputs. However, replacing a frequently repeating pattern will result in loss of context due to the excessively general semantics of the pattern. Furthermore, due to the lack of ground truth, denoising module often relies on parametric models that still learn distribution from noisy data, which further limits model performance. In this paper, we propose Triple-d, a technique for high-quality data creation through adaptive pattern replacement and a scalable non-parametric model. Specifically, we formulate the adaptive pattern replacement task as a maximum-profit bipartite graph problem and propose an approximation algorithm as a solution. Additionally, we design a non-parametric model with scalable instance normalization to efficiently estimate and eliminate the influence of each dimension in neighbors. Extensive experiments in the denoising task and a downstream relation extraction task on real-world datasets demonstrate the superior effectiveness and efficiency of Triple-d, highlighting its potential to improve the performance for high-quality data creation.11Corresponding author: Yongqi Zhang. Lei Chen 0002, Kai Chen 0005 |
ICDE | 4 |
| 2024 | Optimizing Privacy, Utility, and Efficiency in a Constrained Multi-Objective Federated Learning FrameworkabstractConventionally, federated learning aims to optimize a single objective, typically the utility. However, for a federated learning system to be trustworthy, it needs to simultaneously satisfy multiple objectives, such as maximizing model performance, minimizing privacy leakage and training costs, and being robust to malicious attacks. Multi-Objective Optimization (MOO) aiming to optimize multiple conflicting objectives simultaneously is quite suitable for solving the optimization problem of Trustworthy Federated Learning (TFL). In this article, we unify MOO and TFL by formulating the problem of constrained multi-objective federated learning (CMOFL). Under this formulation, existing MOO algorithms can be adapted to TFL straightforwardly. Different from existing CMOFL algorithms focusing on utility, efficiency, fairness, and robustness, we consider optimizing privacy leakage along with utility loss and training cost, the three primary objectives of a TFL system. We develop two improved CMOFL algorithms based on NSGA-II and PSL, respectively, to effectively and efficiently find Pareto optimal solutions and provide theoretical analysis on their convergence. We design quantitative measurements of privacy leakage, utility loss, and training cost for three privacy protection mechanisms: Randomization, BatchCrypt (an efficient homomorphic encryption), and Sparsification. Empirical experiments conducted under the three protection mechanisms demonstrate the effectiveness of our proposed algorithms. Yan Kang 0001, Hanlin Gu, Xingxing Tang, Yuanqin He, Yuzhu Zhang, Jinnan He, Yuxing Han 0001, Lixin Fan, Kai Chen 0005, Qiang Yang 0001 |
ACM Trans. Intell. Syst. Technol. | 9 |
| 2024 | A Game-theoretic Framework for Privacy-preserving Federated LearningabstractIn federated learning, benign participants aim to optimize a global model collaboratively. However, the risk of privacy leakage cannot be ignored in the presence of semi-honest adversaries. Existing research has focused either on designing protection mechanisms or on inventing attacking mechanisms. While the battle between defenders and attackers seems never-ending, we are concerned with one critical question: Is it possible to prevent potential attacks in advance? To address this, we propose the first game-theoretic framework that considers both FL defenders and attackers in terms of their respective payoffs, which include computational costs, FL model utilities, and privacy leakage risks. We name this game the federated learning privacy game (FLPG), in which neither defenders nor attackers are aware of all participants’ payoffs. To handle the incomplete information inherent in this situation, we propose associating the FLPG with an oracle that has two primary responsibilities. First, the oracle provides lower and upper bounds of the payoffs for the players. Second, the oracle acts as a correlation device, privately providing suggested actions to each player. With this novel framework, we analyze the optimal strategies of defenders and attackers. Furthermore, we derive and demonstrate conditions under which the attacker, as a rational decision-maker, should always follow the oracle’s suggestion not to attack . Xiaojin Zhang 0002, Lixin Fan, Wenjie Li 0008, Kai Chen 0005, Qiang Yang 0001 |
ACM Trans. Intell. Syst. Technol. | 5 |
| 2024 | A Meta-Learning Framework for Tuning Parameters of Protection Mechanisms in Trustworthy Federated LearningabstractTrustworthy federated learning typically leverages protection mechanisms to guarantee privacy. However, protection mechanisms inevitably introduce utility loss or efficiency reduction while protecting data privacy. Therefore, protection mechanisms and their parameters should be carefully chosen to strike an optimal tradeoff amongprivacy leakage,utility loss, andefficiency reduction. To this end, federated learning practitioners need tools to measure the three factors and optimize the tradeoff between them to choose the protection mechanism that is most appropriate to the application at hand. Motivated by this requirement, we propose a framework that (1) formulates trustworthy federated learning as a problem of finding a protection mechanism to optimize the tradeoff among privacy leakage, utility loss, and efficiency reduction and (2) formally defines bounded measurements of the three factors. We then propose a meta-learning algorithm to approximate this optimization problem and find optimal protection parameters for representative protection mechanisms, including randomization, homomorphic encryption, secret sharing, and compression. We further design estimation algorithms to quantify these found optimal protection parameters in a practical horizontal federated learning setting and provide a theoretical analysis of the estimation error. Xiaojin Zhang 0002, Yan Kang 0001, Lixin Fan, Kai Chen 0005, Qiang Yang 0001 |
ACM Trans. Intell. Syst. Technol. | 4 |
| 2024 | A Survey for Federated Learning Evaluations: Goals and MeasuresabstractEvaluation is a systematic approach to assessing how well a system achieves its intended purpose. Federated learning (FL) is a novel paradigm for privacy-preserving machine learning that allows multiple parties to collaboratively train models without sharing sensitive data. However, evaluating FL is challenging due to its interdisciplinary nature and diverse goals, such as utility, efficiency, and security. In this survey, we first review the major evaluation goals adopted in the existing studies and then explore the evaluation metrics used for each goal. We also introduceFedEval, an open-source platform that provides a standardized and comprehensive evaluation framework for FL algorithms in terms of their utility, efficiency, and security. Finally, we discuss several challenges and future research directions for FL evaluation. Di Chai, Leye Wang, Liu Yang 0008, Junxue Zhang 0001, Kai Chen 0005, Qiang Yang 0001 |
IEEE Trans. Knowl. Data Eng. | 5 |
| 2023 | Transferable Graph Structure Learning for Graph-based Traffic Forecasting Across CitiesabstractGraph-based deep learning models are powerful in modeling spatio-temporal graphs for traffic forecasting. In practice, accurate forecasting models rely on sufficient traffic data, which may not be accessible in real-world applications. To address this problem, transfer learning methods are designed to transfer knowledge from the source graph with abundant data to the target graph with limited data. However, existing methods adopt pre-defined graph structures for knowledge extraction and transfer, which may be noisy or biased and negatively impact the performance of knowledge transfer. To address the problem, we propose TransGTR, a transferable structure learning framework for traffic forecasting that jointly learns and transfers the graph structures and forecasting models across cities. TransGTR consists of a node feature network, a structure generator, and a forecasting model. We train the node feature network with knowledge distillation to extract city-agnostic node features, such that the structure generator, taking the node features as inputs, can be transferred across both cities. Furthermore, we train the structure generator via a temporal decoupled regularization, such that the spatial features learned with the generated graphs share similar distributions across cities and thus facilitate knowledge transfer for the forecasting model. We evaluate TransGTR on real-world traffic speed datasets, where under a fair comparison, TransGTR outperforms state-of-the-art baselines by up to 5.4%. Yilun Jin, Kai Chen 0005, Qiang Yang 0001 |
KDD | 2 |
| 2023 | TrustGeo: Uncertainty-Aware Dynamic Graph Learning for Trustworthy IP GeolocationabstractThe rising popularity of online social network services has attracted a lot of research focusing on mining various user patterns. Among them, accurate IP geolocation is essential for a plethora of location-aware applications. However, despite extensive research efforts and significant advances, the "accurate and reliable'' desideratum is yet to be achieved at a higher quality level. This work presents a graph neural network (GNN)-based model, called TrustGeo, for trustworthy street-level IP geolocation. A distinct and important aspect of TrustGeo is the incorporation of sources of uncertainty in the learning process. The results of our extensive experimental evaluations on three real-world datasets demonstrate the superiority of our framework in significantly improving the accuracy and trustworthiness of street-level IP geolocation. Our code and datasets are available at https://github.com/ICDM-UESTC/TrustGeo. Wenxin Tai, Bin Chen 0030, Fan Zhou 0002, Ting Zhong, Goce Trajcevski, Yong Wang 0046, Kai Chen 0005 |
KDD | 7 |
| 2023 | RIPGeo: Robust Street-Level IP GeolocationabstractIP geolocation refers to the process of determining the geographic locations of Internet Protocol (IP) addresses, which is important for mobile computing and spatial data management. Despite extensive research efforts, a client-independent geolocation service with high accuracy and reliability has not yet been developed. This paper presents a graph neural network (GNN) model, dubbed RIPGeo, for robust street-level IP geolocation. Three factors that affect data quality are identified, and the importance of considering data quality in algorithm development is emphasized. Two novel self-supervised perturbational training strategies are proposed to enhance the generalization and robustness of the model. A multi-task learning framework is introduced to solve the homogenized representation problem caused by perturbational training, demonstrating much more efficiency than prevailing solutions. Theoretical analysis and experimental results demonstrate the superiority of our framework in significantly improving the accuracy and stability of street-level IP geolocation. Wenxin Tai, Bin Chen 0030, Ting Zhong, Yong Wang 0046, Kai Chen 0005, Fan Zhou 0002 |
MDM | 5 |
| 2023 | Scalable and Efficient Full-Graph GNN Training for Large GraphsabstractGraph Neural Networks (GNNs) have emerged as powerful tools to capture structural information from graph-structured data, achieving state-of-the-art performance on applications such as recommendation, knowledge graph, and search. Graphs in these domains typically contain hundreds of millions of nodes and billions of edges. However, previous GNN systems demonstrate poor scalability because large and interleaved computation dependencies in GNN training cause significant overhead in current parallelization methods. We present G3, a distributed system that can efficiently train GNNs over billion-edge graphs at scale. G3 introduces GNN hybrid parallelism which synthesizes three dimensions of parallelism to scale out GNN training by sharing intermediate results peer-to-peer in fine granularity, eliminating layer-wise barriers for global collective communication or neighbor replications as seen in prior works. G3 leverages locality-aware iterative partitioning and multi-level pipeline scheduling to exploit acceleration opportunities by distributing balanced workload among workers and overlapping computation with communication in both inter-layer and intra-layer training processes. We show via a prototype implementation and comprehensive experiments that G3 can achieve as much as 2.24x speedup in a 16-node cluster, and better final accuracy over prior works. Xinchen Wan, Kaiqiang Xu, Xudong Liao, Yilun Jin, Kai Chen 0005, Xin Jin 0008 |
Proc. ACM Manag. Data | 5 |
| 2023 | Federated Clique Percolation for Privacy-preserving Overlapping Community DetectionabstractCommunity structure is a typical characteristic of complex networks. Finding communities in complex networks has many important applications, such as the advertisement and recommendation based on social networks and the discovery of new protein molecules in biological networks, which make it a hot topic in the field of complex network analysis. With the increasing concerns about the leakage of personal privacy, discovering communities spread across the local networks owned by multiple participants accurately while preserving each participant’s privacy has become an emerging challenge in distributed community detection. In this article, we propose a general federated graph learning model for privacy-preserving distributed graph learning and develop two federated clique percolation algorithms (CPAs) based on it to discover overlapping communities distributed across multiple participants’ local networks without disclosing any participant’s network privacy. Homomorphic encryption and hash operation are used in combination to protect the privacy of the vertices and edges of each local network. Furthermore, vertex attributes are involved in the calculation of clique similarity and clique percolation when dealing with attributed networks. The experimental results on real-world and artificial datasets demonstrate that the proposed algorithms achieve identical results to those of their stand-alone counterparts and more than 200% higher accuracy than the simple distributed CPAs without federating learning. Kun Guo 0003, Wenzhong Guo, Enjie Ye, Yutong Fang, Jiachen Zheng, Ximeng Liu, Kai Chen 0005 |
ACM Trans. Intell. Syst. Technol. | 7 |
| 2023 | No Free Lunch Theorem for Security and Utility in Federated LearningabstractIn a federated learning scenario where multiple parties jointly learn a model from their respective data, there exist two conflicting goals for the choice of appropriate algorithms. On one hand, private and sensitive training data must be kept secure as much as possible in the presence of semi-honest partners; on the other hand, a certain amount of information has to be exchanged among different parties for the sake of learning utility. Such a challenge calls for the privacy-preserving federated learning solution, which maximizes the utility of the learned model and maintains a provable privacy guarantee of participating parties’ private data. This article illustrates a general framework that (1) formulates the trade-off between privacy loss and utility loss from a unified information-theoretic point of view, and (2) delineates quantitative bounds of the privacy-utility trade-off when different protection mechanisms including randomization, sparsity, and homomorphic encryption are used. It was shown that in general there is no free lunch for the privacy-utility trade-off , and one has to trade the preserving of privacy with a certain degree of degraded utility. The quantitative analysis illustrated in this article may serve as the guidance for the design of practical federated learning algorithms. Xiaojin Zhang 0002, Hanlin Gu, Lixin Fan, Kai Chen 0005, Qiang Yang 0001 |
ACM Trans. Intell. Syst. Technol. | 4 |
| 2023 | Trading Off Privacy, Utility, and Efficiency in Federated LearningabstractFederated learning (FL) enables participating parties to collaboratively build a global model with boosted utility without disclosing private data information. Appropriate protection mechanisms have to be adopted to fulfill the opposing requirements in preserving privacy and maintaining high model utility . In addition, it is a mandate for a federated learning system to achieve high efficiency in order to enable large-scale model training and deployment. We propose a unified federated learning framework that reconciles horizontal and vertical federated learning. Based on this framework, we formulate and quantify the trade-offs between privacy leakage, utility loss, and efficiency reduction, which leads us to the No-Free-Lunch (NFL) theorem for the federated learning system. NFL indicates that it is unrealistic to expect an FL algorithm to simultaneously provide excellent privacy, utility, and efficiency in certain scenarios. We then analyze the lower bounds for the privacy leakage, utility loss, and efficiency reduction for several widely-adopted protection mechanisms, including Randomization , Homomorphic Encryption , Secret Sharing, and Compression . Our analysis could serve as a guide for selecting protection parameters to meet particular requirements. Xiaojin Zhang 0002, Yan Kang 0001, Kai Chen 0005, Lixin Fan, Qiang Yang 0001 |
ACM Trans. Intell. Syst. Technol. | 3 |
| 2023 | Exploring the Generalizability of Spatio-Temporal Traffic Prediction: Meta-Modeling and an Analytic FrameworkabstractThe Spatio-Temporal Traffic Prediction (STTP) problem is a classical problem with plenty of prior research efforts that benefit from traditional statistical learning and recent deep learning approaches. While STTP can refer to many real-world problems, most existing studies focus on quite specific applications, such as the prediction of taxi demand, ridesharing order, traffic speed, and so on. This hinders the STTP research as the approaches designed for different applications are hardly comparable, and thus how an application-driven approach can be generalized to other scenarios is unclear. To fill in this gap, this paper makes three efforts: (i) we propose an analytic framework, called STAnalytic, to qualitatively investigate STTP approaches regarding their design considerations on various spatial and temporal factors, aiming to make different application-driven approaches comparable; (ii) we design a spatio-temporal meta-model, called STMeta, which can flexibly integrate generalizable temporal and spatial knowledge identified by STAnalytic, (iii) we build an STTP benchmark platform including ten real-life datasets with five scenarios to quantitatively measure the generalizability of STTP approaches. In particular, we implement STMeta with different deep learning techniques, and STMeta demonstrates better generalizability than state-of-the-art approaches by achieving lower prediction error on average across all the datasets. Leye Wang, Di Chai, Xuanzhe Liu, Liyue Chen, Kai Chen 0005 |
IEEE Trans. Knowl. Data Eng. | 5 |
| 2023 | Unified route representation learning for multi-modal transportation recommendation with spatiotemporal pre-training
Hao Liu 0026, Jindong Han, Yanjie Fu, Kai Chen 0005, Hui Xiong 0001 |
VLDB J. | 5 |
| 2022 | Urban Region Profiling via Multi-Graph Representation LearningabstractProfiling urban regions is essential for urban analytics and planning. Although existing studies have made great efforts to learn urban region representation from multi-source urban data, there are still limitations on modelling local-level signals, developing an effective yet integrated fusion framework, and performing well in regions with high variance socioeconomic attributes. Thus, we propose a multi-graph representation learning framework, called Region2Vec, for urban region profiling. Specifically, except that human mobility is encoded for inter-region relations, geographic neighborhood is introduced for capturing geographical contextual information while POI side information is adopted for representing intra-region information. Then, graphs are used to capture accessibility, vicinity, and functionality correlations among regions. An encoder-decoder multi-graph fusion module is further proposed to jointly learn comprehensive representations. Experiments on real-world datasets show that Region2Vec can be employed in three applications and outperforms all state-of-the-art baselines. Particularly, Region2Vec has better performance than previous studies in regions with high variance socioeconomic attributes. Korris Fu-Lai Chung, Kai Chen 0005 |
CIKM | 3 |
| 2022 | Practical Lossless Federated Singular Vector Decomposition over Billion-Scale DataabstractWith the enactment of privacy-preserving regulations, e.g., GDPR, federated SVD is proposed to enable SVD-based applications over different data sources without revealing the original data. However, many SVD-based applications cannot be well supported by existing federated SVD solutions. The crux is that these solutions, adopting either differential privacy (DP) or homomorphic encryption (HE), suffer from accuracy loss caused by unremovable noise or degraded efficiency due to inflated data. Di Chai, Leye Wang, Junxue Zhang 0001, Liu Yang 0008, Shuowei Cai, Kai Chen 0005, Qiang Yang 0001 |
KDD | 6 |
| 2022 | Selective Cross-City Transfer Learning for Traffic Prediction via Source City Region Re-WeightingabstractDeep learning models have been demonstrated powerful in modeling complex spatio-temporal data for traffic prediction. In practice, effective deep traffic prediction models rely on large-scale traffic data, which is not always available in real-world scenarios. To alleviate the data scarcity issue, a promising way is to use cross-city transfer learning methods to fine-tune well-trained models from source cities with abundant data. However, existing approaches overlook the divergence between source and target cities, and thus, the trained model from source cities may contain noise or even harmful source knowledge. To address the problem, we propose CrossTReS, a selective transfer learning framework for traffic prediction that adaptively re-weights source regions to assist target fine-tuning. As a general framework for fine-tuning-based cross-city transfer learning, CrossTReS consists of a feature network, a weighting network, and a prediction model. We train the feature network with node- and edge-level domain adaptation techniques to learn generalizable spatial features for both source and target cities. We further train the weighting network via source-target joint meta-learning such that source regions helpful to target fine-tuning are assigned high weights. Finally, the prediction model is selectively trained on the source city with the learned weights to initialize target fine-tuning. We evaluate CrossTReS using real-world taxi and bike data, where under the same settings, CrossTReS outperforms state-of-the-art baselines by up to 8%. Moreover, the learned region weights offer interpretable visualization. Yilun Jin, Kai Chen 0005, Qiang Yang 0001 |
KDD | 2 |
| 2022 | Connecting the Hosts: Street-Level IP Geolocation with Graph Neural NetworksabstractPinpointing the geographic location of an IP address is important for a range of location-aware applications spanning from targeted advertising to fraud prevention. The majority of traditional measurement-based and recent learning-based methods either focus on the efficient employment of topology or utilize data mining to find clues of the target IP in publicly available sources. Motivated by the limitations in existing works, we propose a novel framework named GraphGeo, which provides a complete processing methodology for street-level IP geolocation with the application of graph neural networks. It incorporates IP hosts knowledge and kinds of neighborhood relationships into the graph to infer spatial topology for high-quality geolocation prediction. We explicitly consider and alleviate the negative impact of uncertainty caused by network jitter and congestion, which are pervasive in complicated network environments. Extensive evaluations across three large-scale real-world datasets demonstrate that GraphGeo significantly reduces the geolocation errors compared to the state-of-the-art methods. Moreover, the proposed framework has been deployed on the web platform as an online service for 6 months. Zhiyuan Wang 0006, Fan Zhou 0002, Wenxuan Zeng, Goce Trajcevski, Chunjing Xiao, Yong Wang 0046, Kai Chen 0005 |
KDD | 7 |
| 2022 | Efficient Federated Matrix Factorization Against Inference AttacksabstractRecommender systems typically require the revelation of users’ ratings to the recommender server, which will subsequently use these ratings to provide personalized services. However, such revelations make users vulnerable to a broader set of inference attacks, allowing the recommender server to learn users’ private attributes, e.g., age and gender. Therefore, in this paper, we propose an efficient federated matrix factorization method that protects users against inference attacks. The key idea is that we obfuscate one user’s rating to another such that the private attribute leakage is minimized under the given distortion budget, which bounds the recommending loss and overhead of system efficiency. During the obfuscation, we apply differential privacy to control the information leakage between the users. We also adopt homomorphic encryption to protect the intermediate results during training. Our framework is implemented and tested on real-world datasets. The result shows that our method can reduce up to 16.7% of inference attack accuracy compared to using no privacy protections. Di Chai, Leye Wang, Kai Chen 0005, Qiang Yang 0001 |
ACM Trans. Intell. Syst. Technol. | 3 |
| 2022 | Improving Availability of Vertical Federated Learning: Relaxing Inference on Non-overlapping DataabstractVertical Federated Learning (VFL) enables multiple parties to collaboratively train a machine learning model over vertically distributed datasets without data privacy leakage. However, there is a limitation of the current VFL solutions: current VFL models fail to conduct inference on non-overlapping samples during inference. This limitation seriously damages the VFL model’s availability because, in practice, overlapping samples may only take up a small portion of the whole data at each party which means a large part of inference tasks will fail. In this article, we propose a novel VFL framework which enables federated inference on non-overlapping data. Our framework regards the distributed features as privileged information which is available in the training period but disappears during inference. We distill the knowledge of such privileged features and transfer them to the parties’ local model which only processes local features. Furthermore, we adopt Oblivious Transfer (OT) to preserve data ID privacy during training and inference. Empirically, we evaluate the model on the real-world dataset collected from Criteo and Taobao. Besides, we also provide a security analysis of the proposed framework. Zhenghang Ren, Liu Yang 0008, Kai Chen 0005 |
ACM Trans. Intell. Syst. Technol. | 3 |
| 2020 | Exploring Clustering of Bandits for Online Recommendation SystemabstractCluster-of-bandit policy leverages contextual bandits in a collaborative filtering manner and aids personalized services in the online recommendation system (RecSys). When facing insufficient observations, the cluster-of-bandit policy could achieve more outstanding performance because of knowledge sharing. Cluster-of-bandit policy aims to maximize the cumulative feedback, e.g., clicks, from users. Nevertheless, in the way of their goal exist two kinds of uncertainties. First, cluster-of-bandit algorithms make recommendations according to their uncertain estimation of user interests. Second, cluster-of-bandit algorithms transfer relevant knowledge upon uncertain and noisy user clusters. Existing algorithms only consider the first one, while leaving the latter one untouched. To address the two challenges together, in this paper, we propose the ClexB policy for online RecSys. On the one hand, ClexB estimates user clustering more accurately and with less uncertainty via explorable-clustering. On the other hand, ClexB also exploits and explores user interests by sharing information within and among user clusters. In summary, ClexB explores knowledge transfer and further aids the inferences about user interests. Besides, we provide extensive empirical experiments on both the synthetic and real-world datasets and regret analysis, further consolidating the superiority of ClexB. Liu Yang 0008, Bo Liu 0015, Leyu Lin, Feng Xia 0006, Kai Chen 0005, Qiang Yang 0001 |
RecSys | 5 |