Demonstration venue · read-only. Every page can be browsed; the buttons that would change it are switched off. Create an account to run TaxoReview on your own data. A simulation from public publication records: the people listed were not invited and did not agree to review.
? Tutorial for this page

Yi He 0007

dblp:65/425-7 · DBLP profile ↗
← Back
68ranked-venue papers
11as first author
58since 2021 · last 2027
0000-0002-5357-6623ORCID · conflict

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

Artificial intelligence and machine learning · 37 · 9 first-author · 30 since 2021Databases, data management, data science and information retrieval · 27 · 4 first-author · 23 since 2021Graphics, computer vision, multimedia, augmented reality and games · 18 · 5 first-author · 15 since 2021Software engineering, systems software and programming languages · 6 · 5 since 2021Applied, interdisciplinary, general and emerging computing · 5 · 1 first-author · 4 since 2021Human-computer interaction and ubiquitous computing · 4 · 4 since 2021Computer networks · 1 · 1 since 2021Security and privacy · 1 · 1 since 2021

Expertise — from the expertise taxonomy: the topics of the expert's papers under the CCF categories. A weight counts papers with recency: 1 for a paper about the topic, 0.3 when the topic is its context, halved every five years.

Artificial intelligence
24 papers
Graph learning · 30% Efficient and distributed learning · 11% Learning paradigms · 9%
Databases, data mining, and information retrieval
18 papers
Data mining · 35% Knowledge graphs · 22% Machine learning and data management · 16%

Topics — the 30 heaviest of 99, each with the papers that count most for it

TopicWeightPapersLastEvidence papers
Machine learning › Learning theory
online learning
2.952024
Online Learning From Evolving Feature Spaces With Deep Variational Models · IEEE Trans. Knowl. Data Eng. 2024
Towards Utilitarian Online Learning - A Review of Online Algorithms in Open Feature Space · IJCAI 2023
Online Deep Learning from Doubly-Streaming Data · ACM Multimedia 2022
Machine learning › Graph learning
graph neural network
2.742025
HGEN: Heterogeneous Graph Ensemble Networks · IJCAI 2025
Topology-aware Neural Flux Prediction Guided by Physics · ICML 2025
GLDL: Graph Label Distribution Learning · AAAI 2024
Machine learning › Efficient and distributed learning
federated learning
2.732026
Federated Latent Factorization of Tensors for Privacy-Preserving Representation Learning to Large-Scale Dynamic Weighted Directed Network · IEEE Trans. Dependable Secur. Comput. 2026
Federated Latent Factor Learning for Privacy-Preserving Spatio-Temporal Signal Recovery · WWW 2026
Privacy-Preserving Multi-Granular Federated Neural Architecture Search - A General Framework · IEEE Trans. Knowl. Data Eng. 2023
Machine learning and data management
online learning
2.042024
Online Semi-supervised Learning with Mix-Typed Streaming Features · AAAI 2023
Online Random Feature Forests for Learning in Varying Feature Spaces · AAAI 2023
Online Learning in Variable Feature Spaces with Mixed Data · ICDM 2021
Machine learning › Graph learning
graph ensemble learning
1.722025
HGEN: Heterogeneous Graph Ensemble Networks · IJCAI 2025
LHGEL: Large Heterogeneous Graph Ensemble Learning using Batch View Aggregation · ICDM 2025
Data mining
anomaly detection
1.722025
Online Outlier Detection in Open Feature Spaces · IEEE Trans. Knowl. Data Eng. 2025
A Novel Sparse Active Online Learning Framework for Fast and Accurate Streaming Anomaly Detection Over Data Streams · IJCAI 2025
Machine learning › Learning paradigms
label distribution learning
1.522024
HGDL: Heterogeneous Graph Label Distribution Learning · NeurIPS 2024
GLDL: Graph Label Distribution Learning · AAAI 2024
Machine learning and data management
feature space evolution
1.222023
Towards Utilitarian Online Learning - A Review of Online Algorithms in Open Feature Space · IJCAI 2023
Online Learning in Variable Feature Spaces with Mixed Data · ICDM 2021
Recommender systems › collaborative filtering
matrix factorization
1.122022
A Posterior-Neighborhood-Regularized Latent Factor Model for Highly Accurate Web Service QoS Prediction · IEEE Trans. Serv. Comput. 2022
A Data-Characteristic-Aware Latent Factor Model for Web Services QoS Prediction · IEEE Trans. Knowl. Data Eng. 2022
Recommender systems
qos prediction
1.122022
A Posterior-Neighborhood-Regularized Latent Factor Model for Highly Accurate Web Service QoS Prediction · IEEE Trans. Serv. Comput. 2022
A Data-Characteristic-Aware Latent Factor Model for Web Services QoS Prediction · IEEE Trans. Knowl. Data Eng. 2022
Machine learning › Probabilistic and Bayesian machine learning › causal inference
causal discovery
1.012026
MARLIN: Multi-Agent Reinforcement Learning for Incremental DAG Discovery · AAAI 2026
Machine learning › Probabilistic and Bayesian machine learning › causal inference › causal discovery
directed acyclic graph learning
1.012026
MARLIN: Multi-Agent Reinforcement Learning for Incremental DAG Discovery · AAAI 2026
Natural language and speech › Language models and text generation › retrieval-augmented generation
graph-based RAG
1.012026
Relink: Constructing Query-Driven Evidence Graph On-the-Fly for GraphRAG · AAAI 2026
Knowledge, reasoning and agents › Knowledge representation and reasoning
knowledge graph
1.012026
Relink: Constructing Query-Driven Evidence Graph On-the-Fly for GraphRAG · AAAI 2026
Machine learning › Reinforcement learning
multi-agent reinforcement learning
1.012026
MARLIN: Multi-Agent Reinforcement Learning for Incremental DAG Discovery · AAAI 2026
Machine learning › Efficient and distributed learning › federated learning
privacy-preserving federated learning
1.012026
Federated Latent Factorization of Tensors for Privacy-Preserving Representation Learning to Large-Scale Dynamic Weighted Directed Network · IEEE Trans. Dependable Secur. Comput. 2026
Natural language and speech › Language models and text generation
retrieval-augmented generation
1.012026
Relink: Constructing Query-Driven Evidence Graph On-the-Fly for GraphRAG · AAAI 2026
Machine learning › Representation and self-supervised learning
tensor decomposition
1.012026
Federated Latent Factorization of Tensors for Privacy-Preserving Representation Learning to Large-Scale Dynamic Weighted Directed Network · IEEE Trans. Dependable Secur. Comput. 2026
Information retrieval › hashing
binary hashing
1.012026
Non-Gradient Hash Factor Learning for High-Dimensional and Incomplete Data Representation Learning · IEEE Trans. Pattern Anal. Mach. Intell. 2026
Knowledge graphs › knowledge graph alignment
entity alignment
1.012026
PSQE: A Theoretical-Practical Approach to Pseudo Seed Quality Enhancement for Unsupervised Multimodal Entity Alignment · KDD (1) 2026
Knowledge graphs › knowledge graph alignment › entity alignment
multi-modal entity alignment
1.012026
PSQE: A Theoretical-Practical Approach to Pseudo Seed Quality Enhancement for Unsupervised Multimodal Entity Alignment · KDD (1) 2026
Data mining
representation learning
1.012026
Non-Gradient Hash Factor Learning for High-Dimensional and Incomplete Data Representation Learning · IEEE Trans. Pattern Anal. Mach. Intell. 2026
Knowledge graphs › knowledge graph alignment › entity alignment
unsupervised entity alignment
1.012026
PSQE: A Theoretical-Practical Approach to Pseudo Seed Quality Enhancement for Unsupervised Multimodal Entity Alignment · KDD (1) 2026
Network measurement and analytics
missing data recovery
1.012026
Federated Latent Factor Learning for Privacy-Preserving Spatio-Temporal Signal Recovery · WWW 2026
Internet of things and sensor networks
wireless sensor network
1.012026
Federated Latent Factor Learning for Privacy-Preserving Spatio-Temporal Signal Recovery · WWW 2026
Machine learning › Learning paradigms
continual learning
0.912025
Metric-Agnostic Continual Learning for Sustainable Group Fairness · AAAI 2025
Machine learning › Graph learning
directed graph
0.912025
Topology-aware Neural Flux Prediction Guided by Physics · ICML 2025
Machine learning › Trustworthy machine learning
fairness
0.912025
Metric-Agnostic Continual Learning for Sustainable Group Fairness · AAAI 2025
Machine learning › Trustworthy machine learning › fairness
group fairness
0.912025
Metric-Agnostic Continual Learning for Sustainable Group Fairness · AAAI 2025
Machine learning › Graph learning
heterogeneous graph learning
0.912025
LHGEL: Large Heterogeneous Graph Ensemble Learning using Batch View Aggregation · ICDM 2025

Methods — techniques the papers use, named apart from their topics

spatio-temporal regularization · 3.0latent factor learning · 3.0ensemble learning · 2.0federated learning · 2.0attention mechanism · 1.6density-based clustering · 1.3RAID-6 code · 1.3tensor decomposition · 1.0query-aware evaluation · 1.0pseudo-seed quality enhancement · 1.0nonlinear activation function · 1.0non-gradient optimization · 1.0multi-agent reinforcement learning · 1.0latent relation instantiation · 1.0graph construction · 1.0factored action space · 1.0discrete differential evolution · 1.0contrastive learning · 1.0
YearPublicationVenuePosition
2027 GC-SAPformer: Gaussian Copula enhanced scale-adaptive pathway transformer for sparse multivariate time series forecasting
Junying Yang, Yi He 0007, Dianlong You, Tangtang Xie, Jia Chen 0018, Di Wu 0056
Expert Syst. Appl.2
2026 Relink: Constructing Query-Driven Evidence Graph On-the-Fly for GraphRAG
abstract
Graph-based Retrieval-Augmented Generation (GraphRAG) mitigates hallucinations in Large Language Models (LLMs) by grounding them in structured knowledge. However, current GraphRAG methods are constrained by a prevailing build-then-reason paradigm, which relies on a static, pre-constructed Knowledge Graph (KG). This paradigm faces two critical challenges. First, the KG's inherent incompleteness often breaks reasoning paths. Second, the graph’s low signal-to-noise ratio introduces distractor facts, presenting query-relevant but misleading knowledge that disrupts the reasoning process. To address these challenges, we argue for a reason-and-construct paradigm and propose Relink, a framework that dynamically builds a query-specific evidence graph. To tackle incompleteness, Relink instantiates required facts from a latent relation pool derived from the original text corpus, repairing broken paths on the fly. To handle misleading or distractor facts, Relink employs a unified, query-aware evaluation strategy that jointly considers candidates from both the KG and latent relations, selecting those most useful for answering the query rather than relying on their pre-existence. This empowers Relink to actively discard distractor facts and construct the most faithful and precise evidence path for each query. Extensive experiments on five Open-Domain Question Answering benchmarks show that Relink achieves significant average improvements of 5.4% in EM and 5.2% in F1 over leading GraphRAG baselines, demonstrating the superiority of our proposed framework.
Manzong Huang, Chenyang Bu, Yi He 0007, Xingrui Zhuo, Xindong Wu 0001
AAAI3
2026 MARLIN: Multi-Agent Reinforcement Learning for Incremental DAG Discovery
abstract
Uncovering causal structures from observational data is crucial for understanding complex systems and making informed decisions. While reinforcement learning (RL) has shown promise in identifying these structures in the form of a directed acyclic graph (DAG), existing methods often lack efficiency, making them unsuitable for online applications. In this paper, we propose MARLIN, an efficient multi-agent RL-based approach for incremental DAG learning. MARLIN uses a DAG generation policy that maps a continuous real-valued space to the DAG space as an intra-batch strategy, then incorporates two RL agents—state-specific and state-invariant—to uncover causal relationships and integrates these agents into an incremental learning framework. Furthermore, the framework leverages a factored action space to enhance parallelization efficiency. Extensive experiments on synthetic and real datasets demonstrate that MARLIN outperforms state-of-the-art methods in terms of both efficiency and effectiveness.
Dong Li 0034, Zhengzhang Chen, Xujiang Zhao, Linlin Yu, Zhong Chen 0003, Yi He 0007, Chen Zhao 0010
AAAI6
2026 PSQE: A Theoretical-Practical Approach to Pseudo Seed Quality Enhancement for Unsupervised Multimodal Entity Alignment
abstract
Multimodal Entity Alignment (MMEA) aims to identify equivalent entities across different data modalities, enabling structural data integration that in turn improves the performance of various large language model applications. To lift the requirement of labeled seed pairs that are difficult to obtain, recent methods shifted to an unsupervised paradigm using pseudo-alignment seeds. However, unsupervised entity alignment in multimodal settings remains underexplored, mainly because the incorporation of multimodal information often results in imbalanced coverage of pseudo-seeds within the knowledge graph. To overcome this, we propose PSQE (Pseudo-Seed Quality Enhancement) to improve the precision and graph coverage balance of pseudo seeds via multimodal information and clustering-resampling. Theoretical analysis reveals the impact of pseudo seeds on existing contrastive learning-based MMEA models. In particular, pseudo seeds can influence the attraction and the repulsion terms in contrastive learning at once, whereas imbalanced graph coverage causes models to prioritize high-density regions, thereby weakening their learning capability for entities in sparse regions. Experimental results validate our theoretical findings and show that PSQE as a plug-and-play module can improve the performance of baselines by considerable margins.
Yunpeng Hong, Chenyang Bu, Yi He 0007, Di Wu 0056, Xindong Wu 0001
KDD (1)4
2026 Federated Latent Factor Learning for Privacy-Preserving Spatio-Temporal Signal Recovery
abstract
Wireless sensor network (WSNs) stands out as a burgeoning and promising domain in intelligent sensing. Owing to various factors such as sudden sensor malfunctions or deliberate shutdown of partial nodes to save energy, the collected sensing signals from WSNs commonly have massive missing data, leading to adverse effects on subsequent analysis or decision-making. Latent factor learning (LFL) has proven to be highly effective in recovering the missing data for WSNs. However, the existing LFL models require the collected sensing signals to be maintained in one central place like a central server, which is becoming unacceptable for data owners who are getting increasingly privacy-sensitive. To address this issue, this paper innovatively proposes a f ederated l atent f actor l earning (FLFL) model for privacy-preserving spatio-temporal signal recovery. Its main idea is two-fold: 1) it designs a sensor-level federated learning framework based on LFL, where each sensor only needs to upload gradient information rather than raw data for training a privacy-preserving recovery model, and 2) it incorporates the spatio-temporal correlation into the designed federated learning framework as the regularization constraint to improve its recovery accuracy. With such designs, FLFL can not only accurately recover the missing data of WSNs but also ensure data owners' privacy-preserving of raw data. To evaluate the proposed FLFL model, extensive experiments have been conducted on four real-world WSNs datasets. The results demonstrate that FLFL significantly outperforms five state-of-the-art federated signal recovery models in terms of recovery accuracy with privacy-preserving.
Chengjun Yu, Di Wu 0056, Yi He 0007, Jia Chen 0018, Xin Luo 0001
WWW3
2026 SADD-RFCO: semi-supervised anomalous data detection based on random forest with co-training
Song Deng, Mengfei Sun, Lei Duan, Yi He 0007
Knowl. Inf. Syst.4
2026 A survey on computational pathology foundation models: datasets, adaptation strategies, and evaluation tasks
abstract
Abstract Computational pathology foundation models (CPathFMs) have emerged as a powerful approach for analyzing histopathological data, leveraging self-supervised learning to extract robust feature representations from unlabeled whole-slide images. These models, categorized into uni-modal and multi-modal frameworks, have demonstrated promise in automating complex pathology tasks such as segmentation, classification, and biomarker discovery. However, the development of CPathFMs presents significant challenges, such as limited data accessibility, high variability across datasets, the necessity for domain-specific adaptation, and the lack of standardized evaluation benchmarks. This survey provides a comprehensive review of CPathFMs in computational pathology, focusing on pre-training datasets, adaptation strategies, and evaluation tasks. We analyze key techniques, such as contrastive learning, masked image modeling and multi-modal integration, and highlight existing gaps in current research. Finally, we explore future directions from four perspectives for advancing CPathFMs. This survey serves as a valuable resource for researchers, clinicians, and AI practitioners, guiding the advancement of CPathFMs toward robust and clinically applicable AI-driven pathology solutions.
Dong Li 0034, Guihong Wan, Xintao Wu, Yi He 0007, Zhong Chen 0003, Ajit Johnson Nirmal, Christine G. Lian, Peter K. Sorger, Yevgeniy R. Semenov, Chen Zhao 0010
Knowl. Inf. Syst.5
2026 Non-Gradient Hash Factor Learning for High-Dimensional and Incomplete Data Representation Learning
abstract
High-dimensional and incomplete (HDI) data are ubiquitous in various Big Data-related industrial applications, such as drug innovation and recommender systems. Hash-learning is the most efficient representation learning approach to extract hidden information from HDI data owing to its fast reasoning and low storage. However, an existing hash learning approach commonly employs gradient-based optimization techniques to address the discrete objective caused by the binary nature of hash factors, where the Quantization (i.e., quantizing the real values to binary codes) loss is inevitable, resulting in accuracy loss when representing HDI data. Motivated by these critical and vital issues, this paper proposes a non-gradient hash factor (NGHF) model with three-fold ideas: a) innovating a discrete differential evolution (DDE) algorithm able to simulate the continuous optimization via disabling bits of binary codes based on the projected Hamming dissimilarity, thus enabling an effective discrete optimizer, b) applying the proposed DDE algorithm to directly optimize the discrete learning objective of NGHF defined on HDI data, thereby facilitating its efficient and precise training without any Quantization loss, and c) theoretically proving the convergence of NGHF. As such, NGHF possesses high representation learning ability comparable to that of a real-valued model, making it able to achieve precise binary representation to HDI data. Extensive experimental results on nine real-world datasets demonstrate that NGHF significantly outperforms eight state-of-the-art hash learning models. Moreover, its accuracy is amazingly comparable to that of a real-valued model for HDI data representation learning.
Di Wu 0056, Yi He 0007, Xin Luo 0001, Xinbo Gao 0001
IEEE Trans. Pattern Anal. Mach. Intell.3
2026 Federated Latent Factorization of Tensors for Privacy-Preserving Representation Learning to Large-Scale Dynamic Weighted Directed Network
abstract
Large-scale dynamic weighted directed network (DWDN) is commonly utilized to illustrate the temporal interactions between nodes in numerous applications. Latent factorization of tensors (LFT) is a typical representation learning approach to extract the desired knowledge from a DWDN via low-rank tensor embedding. However, an existing LFT approach requires the target DWDN to be maintained in one central place like a central server, which is becoming unacceptable for users who are getting increasingly privacy-sensitive. To address this vital issue, this paper innovatively proposes a federated latent factorization of tensors (FLFT) model. It can perform accurate and privacy-preserving representation learning to a DWDN based on four-fold ideas: 1) establishing a data-density-oriented federated learning framework to enable different users to efficiently and cooperatively build a shared LFT model with keeping raw data privacy, 2) incorporating the linear biases into the local training of each user to eliminate the personalized biases or local fluctuations, 3) adopting an effective hybrid filling strategy to further protect each user's private interaction information, and 4) designing a customized nonlinear activation function to capture the nonlinear characteristics of users' interactions. Extensive experiments on four DWDNs collected from industrial applications validate that FLFT demonstrates a notable increase in accuracy compared with state-of-the-art federated and non-federated learning approaches in representing a DWDN. The source code of the proposed FLFT model is shared at the following link:https://github.com/wudi1989/FLFT.
Di Wu 0056, Shuai Zhong, Yi He 0007, Xin Luo 0001, Xinbo Gao 0001
IEEE Trans. Dependable Secur. Comput.3
2026 Multimetric Autoencoder for Representing High-Dimensional and Incomplete Data
abstract
High-dimensional and incomplete (HDI) data commonly arise in many complex application scenarios, such as bioinformatics and recommender systems. Deep neural networks (DNNs) exhibit cutting-edge performance in representing HDI data due to their formidable capacity for nonlinear learning. However, previous research primarily concentrates on single-metric-focused models utilizing fixed and exclusive$L_{2}$-norm-based strategies for both loss and regularize terms. Such strategies limit the model’s ability to effectively learn from diverse and heterogeneous HDI data. Recognizing this limitation, this article presents the multimetric autoencoder (MMA) with the following twofold ideas: 1) utilizing multiple$L_{p}$-norms to create four distinct autoencoders, each defining a unique metric representation space with diverse regularize and loss characteristics; 2) integrating these four diverse autoencoders by using a customized, self-adjusting weighting strategy. This innovative approach enhances the model’s capacity to handle heterogeneous and inclusive HDI data effectively, addressing the limitations of previous studies. The theoretical analysis supports the effectiveness of the MMA in harnessing the benefits of diverse multimetric spaces. In the experiments, the MMA is evaluated on six real HDI datasets. The experimental results reveal that the MMA outperforms seven contemporary models in effectively representing HDI data.
Di Wu 0056, Cheng Liang 0003, Yi He 0007, Yan Qiao 0004, Xin Luo 0001
IEEE Trans. Syst. Man Cybern. Syst.3
2025 Metric-Agnostic Continual Learning for Sustainable Group Fairness
abstract
Group Fairness-aware Continual Learning (GFCL) aims to eradicate discriminatory predictions against certain demographic groups in a sequence of diverse learning tasks. This paper explores an even more challenging GFCL problem – how to sustain a fair classifier across a sequence of tasks with covariate shifts and unlabeled data. We propose the MacFRL solution, with its key idea to optimize the sequence of learning tasks. We hypothesize that high-confident learning can be enabled in the optimized task sequence, where the classifier learns from a set of prioritized tasks to glean knowledge, thereby becoming more capable to handle the tasks with substantial distribution shifts that were originally deferred. Theoretical and empirical studies substantiate that MacFRL excels among its GFCL competitors in terms of prediction accuracy and group fair-ness metrics.
Heng Lian 0001, Chen Zhao 0010, Zhong Chen 0003, Xingquan Zhu 0001, My T. Thai, Yi He 0007
AAAI6
2025 Federated Projection Hamming Hash Latent Factor Learning Model
abstract
Federated Learning (FL) offers a means to train models collaboratively across decentralized data sources while preserving privacy. However, prevalent real-valued FL models face challenges with high computational and communication costs, along with potential privacy breaches. Hash Learning, by converting real values to binary codes, presents a solution but is traditionally limited by its lack of privacy considerations and failure to account for the varying importance of bit positions, resulting in information loss. We propose the, Federated Projection-Hamming Hash Latent Factor (FPHLF) model, which addresses three key challenges in traditional FL: computational costs, storage overhead, and privacy risks. FPHLF innovates by a) Utilizing Projected Hamming Distance to enhance binary code representation by recognizing bit significance; b) Implementing efficient bitwise operations to minimize FL's computational and storage demands; and c) Opting for binary update direction matrices over real-valued gradients to reduce communication costs and strengthen privacy. FPHLF achieves high accuracy with reduced computational, storage, and communication expenses, and mitigates privacy risks. Experiments on four real-world datasets show FPHLF outperforms existing hashing and FL methods in accuracy, efficiency, and security. The source code, accessible at https://github.com/LiCaCO3/FPHLF.
Di Wu 0056, Yi He 0007, Xin Luo 0001
CSCWD3
2025 SciTableQA: A Question-Answering Benchmark for Complex Scientific Tables
Kehinde Ajayi, Yi He 0007, Matthew Maisonave, Kris SeekFord, Jian Wu 0006
TPDL2
2025 Uncertainty-Aware Complex Scientific Table Data Extraction
Kehinde Ajayi, Yi He 0007, Jian Wu 0006
ICDAR (3)2
2025 LHGEL: Large Heterogeneous Graph Ensemble Learning using Batch View Aggregation
abstract
Learning from large heterogeneous graphs presents significant challenges due to the scale of networks, heterogeneity in node and edge types, variations in nodal features, and complex local neighborhood structures. This paper advocates for ensemble learning as a natural solution to this problem, whereby training multiple graph learners under distinct sampling conditions, the ensemble inherently captures different aspects of graph heterogeneity. Yet, the crux lies in combining these learners to meet global optimization objective while maintaining computational efficiency on large-scale graphs. In response, we propose LHGEL, an ensemble framework that addresses these challenges through batch sampling with three key components, namely batch view aggregation, residual attention, and diversity regularization. Specifically, batch view aggregation samples subgraphs and forms multiple graph views, while residual attention adaptively weights the contributions of these views to guide node embeddings toward informative subgraphs, thereby improving the accuracy of base learners. Diversity regularization encourages representational disparity across embedding matrices derived from different views, promoting model diversity and ensemble robustness. Our theoretical study demonstrates that residual attention mitigates gradient vanishing issues commonly faced in ensemble learning. Empirical results on five real heterogeneous networks validate that our LHGEL approach consistently outperforms its state-of-the-art competitors by substantial margin. Codes and datasets are available at https://github.com/Chrisshen12/LHGEL.
Yufei Jin, Yi He 0007, Xingquan Zhu 0001
ICDM3
2025 Link Completion for Dynamic Knowledge Graphs with Entity Drift
abstract
Knowledge Graphs (KGs) often suffer from link incompleteness, limiting their performance in downstream tasks. Knowledge graph completion (KGC) has emerged to complete the missing links between entities, so to integrate new knowledge into existing KGs. Unfortunately, traditional KGC models mainly focus on static KGs with a fixed set of entities and relations, thereby struggling in generalization to dynamic KGs with possible entity drift. In this paper, we propose a Link Completion model for dynamic KGs with Entity Drift (LCED) to solve the issue. Unlike traditional link completion models, LCED can generate accurate embeddings for drifted entities, without imposing any assumption on their time or rule characteristic. The drifted entities are incorporated into the KG where their links to existing entities are predicted in an incremental fashion, without retraining the entire KG for computational efficiency. Extensive experiments conducted on benchmark datasets demonstrate the effectiveness of our proposed LCED in link completion for dynamic KGs by outperforming its state-of-the-art competitors. Code and datasets are available at https://github.com/field658/LCED.
Xiulin Zheng, Pei-Pei Li 0001, Yi He 0007, Xindong Wu 0001
ICDM3
2025 Topology-aware Neural Flux Prediction Guided by Physics
abstract
Graph Neural Networks (GNNs) often struggle in preserving high-frequency components of nodal signals when dealing with directed graphs. Such components are crucial for modeling flow dynamics, without which a traditional GNN tends to treat a graph with forward and reverse topologies equal. To make GNNs sensitive to those high-frequency components thereby being capable to capture detailed topological differences, this paper proposes a novel framework that combines 1) explicit difference matrices that model directional gradients and 2) implicit physical constraints that enforce messages passing within GNNs to be consistent with natural laws. Evaluations on two real-world directed graph data, namely, water flux network and urban traffic flow network, demonstrate the effectiveness of our proposal.
Haoyang Jiang, Jindong Wang 0001, Xingquan Zhu 0001, Yi He 0007
ICML4
2025 A Novel Sparse Active Online Learning Framework for Fast and Accurate Streaming Anomaly Detection Over Data Streams
abstract
Online Anomaly Detection (OAD) is critical for identifying rare yet important data points in large, dynamic, and complex data streams. A key challenge lies in achieving accurate and consistent detection of anomalies while maintaining computational and memory efficiency. Conventional OAD approaches, which depend on distributional deviations and static thresholds, struggle with model update delays and catastrophic forgetting, leading to missed detections and high false positive rates. To address these limitations, we propose a novel Streaming Anomaly Detection (SAD) method, grounded in a sparse active online learning framework. Our approach uniquely integrates ℓ1,2-norm sparse online learning with CUR decomposition-based active learning, enabling simultaneous fast feature selection and dynamic instance selection. The efficient CUR decomposition further supports real-time residual analysis for anomaly scoring, eliminating the need for manual threshold settings about temporal data distributions. Extensive experiments on diverse streaming datasets demonstrate SAD's superiority, achieving a 14.06% reduction in detection error rates compared to five state-of-the-art competitors.
Zhong Chen 0003, Yi He 0007, Meikang Qiu
IJCAI2
2025 How to Mitigate Information Loss in Knowledge Graphs for GraphRAG: Leveraging Triple Context Restoration and Query-Driven Feedback
abstract
Knowledge Graph (KG)-augmented Large Language Models (LLMs) have recently propelled significant advances in complex reasoning tasks, thanks to their broad domain knowledge and contextual awareness. Unfortunately, current methods often assume KGs to be complete, which is impractical given the inherent limitations of KG construction and the potential loss of contextual cues when converting unstructured text into entity-relation triples. In response, this paper proposes the Triple Context Restoration and Query-driven Feedback (TCR-QF) framework, which reconstructs the textual context underlying each triple to mitigate information loss, while dynamically refining the KG structure by iteratively incorporating query-relevant missing knowledge. Experiments on five benchmark question-answering datasets substantiate the effectiveness of TCR-QF in KG and LLM integration, where itachieves a 29.1% improvement in Exact Match and a 15.5% improvement in F1 over its state-of-the-art GraphRAG competitors. The code is publicly available at https://github.com/HFUT-DMiC-Lab/TCR-QF.git.
Manzong Huang, Chenyang Bu, Yi He 0007, Xindong Wu 0001
IJCAI3
2025 HGEN: Heterogeneous Graph Ensemble Networks
abstract
This paper presents HGEN that pioneers ensemble learning for heterogeneous graphs. We argue that the heterogeneity in node types, nodal features, and local neighborhood topology poses significant challenges for ensemble learning, particularly in accommodating diverse graph learners. Our HGEN framework ensembles multiple learners through a meta-path and transformation-based optimization pipeline to uplift classification accuracy. Specifically, HGEN uses meta-path combined with random dropping to create Allele Graph Neural Networks (GNNs), whereby the base graph learners are trained and aligned for later ensembling. To ensure effective ensemble learning, HGEN presents two key components:1) a residual-attention mechanism to calibrate allele GNNs of different meta-paths, thereby enforcing node embeddings to focus on more informative graphs to improve base learner accuracy, and 2) a correlation-regularization term to enlarge the disparity among embedding matrices generated from different meta-paths, thereby enriching base learner diversity. We analyze the convergence of HGEN and attest its higher regularization magnitude over simple voting. Experiments on five heterogeneous networks validate that HGEN consistently outperforms its state-of-the-art competitors by substantial margin. Codes are available at https://github.com/Chrisshen12/HGEN.
Yufei Jin, Kaibu Feng, Yi He 0007, Xingquan Zhu 0001
IJCAI4
2025 ℓ1, ∞ Mixed Norm Promoted Row Sparsity for Fast Online CUR Decomposition Learning in Varying Feature Spaces
abstract
Online learning enables effective predictive modeling on complex data streams. To overcome the negative impact of possibly high-dimensional data, sparse online learning (SOL) has been proposed by imposing various sparse constraints to sheer the resultant model structure. However, most existing SOL studies focused on a fixed feature space, whereas in practice the steaming data observations may increment in both quantity and feature dimensions, leading to varying feature spaces. In this paper, we propose a novel ℓ1,∞-mixed norm-based row sparsity SOL algorithm (SOOFS) to handle data streams in varying feature spaces. We empower SOOFS with a tailored online CUR matrix decomposition method based on the promoted row sparsity to actively and adaptively select informative instances in the sliding windows, facilitating stable online performance over time. Empirical results on ten benchmark datasets substantiate the superiority of SOOFS over three state-of-the-art competitors in terms of classification accuracy and model sparsity.
Zhong Chen 0003, Yi He 0007, Di Wu 0056, Wenbin Zhang 0002, Zhiqiang Deng
SDM2
2025 Insulator defect detection from aerial images in adverse weather conditions
Song Deng, Yi He 0007
Appl. Intell.3
2025 Online Learning from Mix-typed, Drifted, and Incomplete Streaming Features
abstract
Online learning, where feature spaces can change over time, offers a flexible learning paradigm that has attracted considerable attention. However, it still faces three significant challenges. First, the heterogeneity of real-world data streams with mixed feature types presents challenges for traditional parametric modeling. Second, data stream distributions can shift over time, causing an abrupt and substantial decline in model performance. Additionally, the time and cost constraints make it infeasible to label every data instance in a supervised setting. To overcome these challenges, we propose a new algorithm Online Learning from Mix-typed, Drifted, and Incomplete Streaming Features (OL-MDISF), which aims to relax restrictions on both feature types, data distribution, and supervision information. Our approach involves utilizing copula models to create a comprehensive latent space, employing an adaptive sliding window for detecting drift points to ensure model stability, and establishing label proximity information based on geometric structural relationships. To demonstrate the model’s efficiency and effectiveness, we provide theoretical analysis and comprehensive experimental results.
Shengda Zhuo, Di Wu 0056, Yi He 0007, Shuqiang Huang, Xindong Wu 0001
ACM Trans. Knowl. Discov. Data3
2025 Online Outlier Detection in Open Feature Spaces
abstract
Outlier detection is essential for data compliance, fraud prevention, and strategic decision-making. Finding outliers relies on study of feature space to find anomalous instances. As the feature dimension increases, it will inevitably complicate the process and hinder the models from finding genuine outliers. In this paper, we investigate an ever-more challenging task, online outlier detection (OOD) problem, where data points to be examined for outlier detection are characterized by two dynamic changes: (1) increasing volume instead of a static set; and (2) evolving feature space instead of a known set. Such instance and feature space dynamics impedes traditional OD techniques reliant on geometric data structure for distinguishing outliers. To aid, we propose a new approach coinedOnline Outlier Detection in Open Feature Spaces, which circumvents this limitation by learning a latent hypersphere representation, respectively positioning regular and anomalous data points inside and outside its boundary. The crux of our approach tailors a reconstruction loss, allowing each data point to be represented as anadditionof its pertinent feature embeddings. Each of these embeddings is updated non-intrusively, championing both efficient and incremental learning of the latent hypersphere. Extensive experiments on twelve benchmark datasets underscore the robustness and superior performance of our method against seven leading counterparts. Code is released inhttps://github.com/X1aoLian/OODOFS.git.
Heng Lian 0001, Yi He 0007, Di Wu 0056, Zhong Chen 0003, Xingquan Zhu 0001, Xindong Wu 0001
IEEE Trans. Knowl. Data Eng.2
2025 Robust Low-Rank Latent Feature Analysis for Spatiotemporal Signal Recovery
abstract
Wireless sensor network (WSN) is an emerging and promising developing area in the intelligent sensing field. Due to various factors like sudden sensors breakdown or saving energy by deliberately shutting down partial nodes, there are always massive missing entries in the collected sensing data from WSNs. Low-rank matrix approximation (LRMA) is a typical and effective approach for pattern analysis and missing data recovery in WSNs. However, existing LRMA-based approaches ignore the adverse effects of outliers inevitably mixed with collected data, which may dramatically degrade their recovery accuracy. To address this issue, this article innovatively proposes a latent feature analysis (LFA) based spatiotemporal signal recovery (STSR) model, named LFA-STSR. Its main idea is twofold: 1) incorporating the spatiotemporal correlation into an LFA model as the regularization constraint to improve its recovery accuracy and 2) aggregating the -norm into the loss part of an LFA model to improve its robustness to outliers. As such, LFA-STSR can accurately recover missing data based on partially observed data mixed with outliers in WSNs. To evaluate the proposed LFA-STSR model, extensive experiments have been conducted on four real-world WSNs datasets. The results demonstrate that LFA-STSR significantly outperforms the related six state-of-the-art models in terms of both recovery accuracy and robustness to outliers.
Di Wu 0056, Zechao Li, Zhikai Yu, Yi He 0007, Xin Luo 0001
IEEE Trans. Neural Networks Learn. Syst.4
2024 GLDL: Graph Label Distribution Learning
abstract
Label Distribution Learning (LDL), as a more general learning setting than generic single-label and multi-label learning, has been commonly used in computer vision and many other applications. To date, existing LDL approaches are designed and applied to data without considering the interdependence between instances. In this paper, we propose a Graph Label Distribution Learning (GLDL) framework, which explicitly models three types of relationships: instance-instance, label-label, and instance-label, to learn the label distribution for networked data. A label-label network is learned to capture label-to-label correlation, through which GLDL can accurately learn label distributions for nodes. Dual graph convolution network (GCN) Co-training with heterogeneous message passing ensures two GCNs, one focusing on instance-instance relationship and the other one targeting label-label correlation, are jointly trained such that instance-instance relationship can help induce label-label correlation and vice versa. Our theoretical study derives the error bound of GLDL. For verification, four benchmark datasets with label distributions for nodes are created using common graph benchmarks. The experiments show that considering dependency helps learn better label distributions for networked data, compared to state-of-the-art LDL baseline. In addition, GLDL not only outperforms simple GCN and graph attention networks (GAT) using distribution loss but is also superior to its variant considering label-label relationship as a static network. GLDL and its benchmarks are the first research endeavors to address LDL for graphs. Code and benchmark data are released for public access.
Yufei Jin, Richard Gao, Yi He 0007, Xingquan Zhu 0001
AAAI3
2024 MKG-FENN: A Multimodal Knowledge Graph Fused End-to-End Neural Network for Accurate Drug-Drug Interaction Prediction
abstract
Taking incompatible multiple drugs together may cause adverse interactions and side effects on the body. Accurate prediction of drug-drug interaction (DDI) events is essential for avoiding this issue. Recently, various artificial intelligence-based approaches have been proposed for predicting DDI events. However, DDI events are associated with complex relationships and mechanisms among drugs, targets, enzymes, transporters, molecular structures, etc. Existing approaches either partially or loosely consider these relationships and mechanisms by a non-end-to-end learning framework, resulting in sub-optimal feature extractions and fusions for prediction. Different from them, this paper proposes a Multimodal Knowledge Graph Fused End-to-end Neural Network (MKGFENN) that consists of two main parts: multimodal knowledge graph (MKG) and fused end-to-end neural network (FENN). First, MKG is constructed by comprehensively exploiting DDI events-associated relationships and mechanisms from four knowledge graphs of drugs-chemical entities, drug-substructures, drugs-drugs, and molecular structures. Correspondingly, a four channels graph neural network is designed to extract high-order and semantic features from MKG. Second, FENN designs a multi-layer perceptron to fuse the extracted features by end-to-end learning. With such designs, the feature extractions and fusions of DDI events are guaranteed to be comprehensive and optimal for prediction. Through extensive experiments on real drug datasets, we demonstrate that MKG-FENN exhibits high accuracy and significantly outperforms state-of-the-art models in predicting DDI events. The source code and supplementary file of this article are available on: https://github.com/wudi1989/MKG-FENN.
Di Wu 0056, Wu Sun, Yi He 0007, Zhong Chen 0003, Xin Luo 0001
AAAI3
2024 ℓ1, 2-Norm and CUR Decomposition based Sparse Online Active Learning for Data Streams with Streaming Features
abstract
Aiming at learning from a sequence of data instances over time, online learning has attracted increasing attention in the big data era. As two important variants, sparse online learning has been extensively explored by facilitating sparse constraints for online models such as truncated gradient, ℓ1-norm regularization, ℓ1-ball projection, and regularized dual averaging; while online active learning aims to build an online prediction model with a limited number of labeled instances, deploying the so called query strategies to select informative instances over time. However, most existing studies consider sparse online learning or online active learning with fixed feature spaces, whereby in real practice the features may be dynamically evolved over time. To the end, we propose a novel unified one-pass online learning framework named OASF for simultaneously online active learning and sparse online learning tailored for data streams described by open feature spaces, where new features can emerge constantly, and old features may be vanished over various time spans. Specifically, we technically develop an effective online CUR matrix decomposition based on the ℓ1,2mixed norm constraint for simultaneously selecting important up-to-date samples in a sliding window and facilitating stable and meaningful features in open feature spaces over time. If the loss function is simultaneously Lipschitz and convex, a sub-linear regret bound of our proposed algorithm is guaranteed with. Extensive experiments that are conducted with multiple streaming datasets have demonstrated the effectiveness of the proposed OASF compared with state-of-the-art online active learning and sparse online learning methods.
Zhong Chen 0003, Yi He 0007, Di Wu 0056, Liudong Zuo, Keren Li, Wenbin Zhang 0002, Zhiqiang Deng
IEEE Big Data2
2024 Learning Gradual Typing Performance
abstract
Gradual typing has emerged as a promising typing discipline for reconciling static and dynamic typing, which have respective strengths and shortcomings. Thanks to its promises, gradual typing has gained tremendous momentum in both industry and academia. A main challenge in gradual typing is that, however, the performance of its programs can often be unpredictable, and adding or removing the type of a a single parameter may lead to wild performance swings. Many approaches have been proposed to optimize gradual typing performance, but little work has been done to aid the understanding of the performance landscape of gradual typing and navigating the migration process (which adds type annotations to make programs more static) to avert performance slowdowns. Motivated by this situation, this work develops a machine-learning-based approach to predict the performance of each possible way of adding type annotations to a program. On top of that, many supports for program migrations could be developed, such as finding the most performant neighbor of any given configuration. Our approach gauges runtime overheads of dynamic type checks inserted by gradual typing and uses that information to train a machine learning model, which is used to predict the running time of gradual programs. We have evaluated our approach on 12 Python benchmarks for both guarded and transient semantics. For guarded semantics, our evaluation results indicate that with only 40 training instances generated from each benchmark, the predicted times for all other instances differ on average by 4% from the measured times. For transient semantics, the time difference ratio is higher but the time difference is often within 0.1 seconds.
Wahiduzzaman Khan, Sheng Chen 0008, Yi He 0007
ECOOP3
2024 Utilitarian Online Learning from Open-World Soft Sensing
abstract
Data-driven soft sensing enables to monitor and control complex industrial processes in real-time. Whereas recent data stream mining algorithms bolster predictive modeling on soft sensing data, which increment in volume and vary in feature dimensions, they operate mainly in closed-world settings, where all class labels must be known beforehand. This is restrictive in practical applications like semiconductor manufacturing, where new wafer defect types emerge dynamically in unforeseeable manners. This study aims to advance online algorithms by allowing learners opt to abstain from make prediction at certain costs. Our key idea is to establish a universal representation space aligning feature dimensions of incoming points while delineating a geometric shape underpinning them. On this shape, we minimize the region spanned by points of known classes through optimizing the trade-off between empirical risk and abstention cost. Theoretical results rationalize our universal representation learning design. We benchmark our approach on six datasets, including one real-world dataset of wafer fault-diagnostics collected through chip manufacturing lines in Seagate. Experimental results substantiate the effectiveness of our proposed approach, demonstrating superior performance over six state-of-the-art rival models. Code and datasets are openly accessible via an anonymous link: https://github.com/X1aoLian/OWSS.
Heng Lian 0001, Yu Huang 0017, Xingquan Zhu 0001, Yi He 0007
ICDM4
2024 End-to-end Graph Learning Approach for Cognitive Diagnosis of Student Tutorial
abstract
Cognitive diagnosis (CD) utilizes students' existing studying records to estimate their mastery of unknown knowledge concepts, which is vital for evaluating their learning abilities. Accurate CD is extremely challenging because CD is associated with complex relationships and mechanisms among students, knowledge concepts, studying records, etc. However, existing approaches loosely consider these relationships and mechanisms by a non-end-to-end learning framework, resulting in sub-optimal feature extractions and fusions for CD. Different from them, this paper innovatively proposes an End-to-end Graph Neural Networks-based Cognitive Diagnosis (EGNN-CD) model. EGNN-CD consists of three main parts: knowledge concept network (KCN), graph neural networks-based feature extraction (GNNFE), and cognitive ability prediction (CAP). First, KCN constructs CD-related interaction by comprehensively extracting physical information from students, exercises, and knowledge concepts. Second, a four-channel GNNFE is designed to extract high-order and individual features from the constructed KCN. Finally, CAP employs a multi-layer perceptron to fuse the extracted features to predict students' learning abilities in an end-to-end learning way. With such designs, the feature extractions and fusions are guaranteed to be comprehensive and optimal for CD. Extensive experiments on three real datasets demonstrate that our EGNN-CD achieves significantly higher accuracy than state-of-the-art models in CD. Our code is available at https://github.com/Yangfl-1112/EGNN-CD.
Fulai Yang, Di Wu 0056, Yi He 0007, Xin Luo 0001
MSN3
2024 HGDL: Heterogeneous Graph Label Distribution Learning
abstract
Label Distribution Learning (LDL) has been extensively studied in IID data applications such as computer vision, thanks to its more generic setting over single-label and multi-label classification. This paper advances LDL into graph domains and aims to tackle a novel and fundamental heterogeneous graph label distribution learning (HGDL) problem. We argue that the graph heterogeneity reflected on node types, node attributes, and neighborhood structures can impose significant challenges for generalizing LDL onto graphs. To address the challenges, we propose a new learning framework with two key components: 1) proactive graph topology homogenization, and 2) topology and content consistency-aware graph transformer. Specifically, the former learns optimal information aggregation between meta-paths, so that the node heterogeneity can be proactively addressed prior to the succeeding embedding learning; the latter leverages an attention mechanism to learn consistency between meta-path and node attributes, allowing network topology and nodal attributes to be equally emphasized during the label distribution learning. By using KL-divergence and additional constraints, \method~delivers an end-to-end solution for learning and predicting label distribution for nodes. Both theoretical and empirical studies substantiate the effectiveness of our HGDL approach. Our code and datasets are available at https://github.com/Listener-Watcher/HGDL.
Yufei Jin, Heng Lian 0001, Yi He 0007, Xingquan Zhu 0001
NeurIPS3
2024 Robust Sparse Online Learning for Data Streams with Streaming Features
abstract
Sparse online learning has received extensive attention during the past few years. Most of existing algorithms that utilize ℓ1-norm regularization or ℓ1-ball projection assume that the feature space is fixed or changes by following explicit constraints. However, this assumption does not always hold in many real applications. Motivated by this observation, we propose a new online learning algorithm tailored for data streams described by open feature spaces, where new features can be occurred, and old features may be vanished over various time spans. Our algorithm named RSOL provides a strategy to adapt quickly to such feature dynamics by encouraging sparse model representation with an ℓ1- and ℓ2 -mixed regularizer. We leverage the proximal operator of the ℓ1,2 -mixed norm and show that our RSOL algorithm enjoys a closed-form solution at each iteration. A sub-linear regret bound of our proposed algorithm is guaranteed with a solid theoretical analysis. Empirical results benchmarked on nine streaming datasets validate the effectiveness of the proposed RSOL method over three state-of-the-art algorithms.
Zhong Chen 0003, Yi He 0007, Di Wu 0056, Huixin Zhan, Victor S. Sheng, Kun Zhang 0012
SDM2
2024 Online Learning From Evolving Feature Spaces With Deep Variational Models
abstract
In this paper, we explore a novel online learning setting, where the online learners are presented with “doubly-streaming” data. Namely, the data instances constantly streaming in are described by feature spaces that over-time evolve, with new features emerging and old features fading away. The main challenge of this problem lies in the fact that the newly emerging features are described by very few samples, resulting in weak learners that tend to make error predictions. A seemingly plausible idea to overcome the challenge is to establish a relationship between the old and new feature spaces, so that an online learner can leverage the knowledge learned from the old features to better the learning performance on the new features. Unfortunately, this idea does not scale up to high-dimensional feature spaces that entail very complex feature interplay. Specifically. a tradeoff between onlineness, which biases shallow learners, and expressiveness, which requires deep models, is inevitable. Motivated by this, we propose a novel paradigm, named Online Learning Deep models from Data of Double Streams (OLD3S), where a shared latent supspace is discovered to summarize information from the old and new feature spaces, building an intermediate feature mapping relationship. A key trait of OLD3S is to treat the model capacity as a learnable semantics, aiming to yield optimal model depth and parameters jointly in accordance with the complexity and non-linearity of the input data streams in an online fashion. To ablate its efficacy and applicability, two variants of OLD3S are proposed namely, OLD-Linear that learns the relationship by a linear function; and OLD-FD learns that two consecutive feature spaces pre-and-post evolution with fixed deep depth. Besides, instead of re-starting the entire learning process from scratch, OLD3S learns multiple newly emerging feature spaces in a lifelong manner, retaining the knowledge from the learned and vanished feature space to enjoy a jump-start of the new features' learning process. Both theoretical analysis and empirical studies supstantiate the viability and effectiveness of our proposed approach. The code is available online atgithub.com/X1aoLian/OLD3S-L.
Heng Lian 0001, Di Wu 0056, Bo-Jian Hou, Jian Wu 0006, Yi He 0007
IEEE Trans. Knowl. Data Eng.5
2024 A Prediction-Sampling-Based Multilayer-Structured Latent Factor Model for Accurate Representation to High-Dimensional and Sparse Data
abstract
Performing highly accurate representation learning on a high-dimensional and sparse (HiDS) matrix is of great significance in a big data-related application such as a recommender system. A latent factor (LF) model is one of the most efficient approaches to the HiDS matrix representation. However, an LF model's representation learning ability relies heavily on an HiDS matrix's known data density, which is extremely low due to numerous missing data entities. To address this issue, this work proposes a prediction-sampling-based multilayer-structured LF (PMLF) model with twofold ideas: 1) constructing a loosely connected multilayered LF architecture to increase the known data density of an input HiDS matrix by generating synthetic data layer by layer and 2) constraining this synthetic data generating process through a random prediction-sampling strategy and nonlinear activations to avoid overfitting. In the experiments, PMLF is compared with six state-of-the-art LF-and deep neural network (DNN)-based models on four HiDS matrices from industrial applications. The results demonstrate that PMLF outperforms its peers in well-balancing prediction accuracy and computational efficiency.
Di Wu 0056, Xin Luo 0001, Yi He 0007, MengChu Zhou
IEEE Trans. Neural Networks Learn. Syst.3
2024 MMLF: Multi-Metric Latent Feature Analysis for High-Dimensional and Incomplete Data
abstract
High-dimensional and incomplete (HDI) data are omnipresent in a variety of Big Data-related applications. Latent feature analysis (LFA) is a typical representation learning method that can extract useful yet latent knowledge from HDI data via low-rank embedding. Existing LFA-based models mostly adopt a single-metric-based modeling strategy, where the representation designed for the embeddingLossfunction is fixed and exclusive. However, real-world HDI data are commonly heterogeneous and have large diverse underlying patterns, making a single-metric-based model cannot represent such HDI data in a comprehensive and unbiased fashion. Motivated by this discovery, this article proposes a multi-metric latent feature (MMLF) model whose ideas are two-fold: 1) two vector spaces and threeLp-norms are simultaneously adopted to develop six LFA variants, each of which possesses a unique merit, and 2) all the variants are aggregated with a tailored, self-adaptive weighting strategy. As such, the proposed MMLF enjoys the merits originated from a set of disparate metric spaces all at once, achieving the comprehensive and unbiased representation of HDI data. Theoretical study guarantees that MMLF attains evident performance gain. Extensive experiments on ten real-world HDI matrices, spanning a wide range of industrial and scientific areas, verify that the proposed MMLF significantly outperforms nine state-of-the-art, shallow and deep counterparts.
Di Wu 0056, Peng Zhang 0125, Yi He 0007, Xin Luo 0001
IEEE Trans. Serv. Comput.3
2023 Online Random Feature Forests for Learning in Varying Feature Spaces
abstract
In this paper, we propose a new online learning algorithm tailored for data streams described by varying feature spaces (VFS), wherein new features constantly emerge and old features may stop to be observed over various time spans. Our proposed algorithm, named Online Random Feature Forests for Feature space Variabilities (ORF3V), provides a strategy to respect such feature dynamics by generating, updating, pruning, as well as online re-weighing an ensemble of what we call feature forests, which are generated and updated based on a compressed and storage efficient representation for each observed feature. We benchmark our algorithm on 12 datasets, including one novel real-world dataset of government COVID-19 responses collected through a crowd-sensing program in Spain. The empirical results substantiate the viability and effectiveness of our ORF3V algorithm and its superior accuracy performance over the state-of-the-art rival models.
Christian Schreckenberger, Yi He 0007, Stefan Lüdtke, Christian Bartelt, Heiner Stuckenschmidt
AAAI2
2023 Online Semi-supervised Learning with Mix-Typed Streaming Features
abstract
Online learning with feature spaces that are not fixed but can vary over time renders a seemingly flexible learning paradigm thus has drawn much attention. Unfortunately, two restrictions prohibit a ubiquitous application of this learning paradigm in practice. First, whereas prior studies mainly assume a homogenous feature type, data streams generated from real applications can be heterogeneous in which Boolean, ordinal, and continuous co-exist. Existing methods that prescribe parametric distributions such as Gaussians would not suffice to model the correlation among such mixtyped features. Second, while full supervision seems to be a default setup, providing labels to all arriving data instances over a long time span is tangibly onerous, laborious, and economically unsustainable. Alas, a semi-supervised online learner that can deal with mix-typed, varying feature spaces is still missing. To fill the gap, this paper explores a novel problem, named Online Semi-supervised Learning with Mixtyped streaming Features (OSLMF), which strives to relax the restrictions on the feature type and supervision information. Our key idea to solve the new problem is to leverage copula model to align the data instances with different feature spaces so as to make their distance measurable. A geometric structure underlying data instances is then established in an online fashion based on their distances, through which the limited labeling information is propagated, from the scarce labeled instances to their close neighbors. Experimental results are documented to evidence the viability and effectiveness of our proposed approach. Code is released in https://github.com/wudi1989/OSLMF.
Di Wu 0056, Shengda Zhuo, Yu Wang 0017, Zhong Chen 0003, Yi He 0007
AAAI5
2023 A Domain Adaptation-based Detector for Cooperative Spectrum Sensing
abstract
Emerging machine learning approaches provide effective solutions in a data-driven manner for cooperative spectrum sensing (CSS). Their success relies on large amounts of labeled training data to capture spectrum characteristics. However, data collection and annotation under dynamic spectrum environments are time-consuming and impractical. To this end, we propose a novel Domain Adaptation-based Detector for CSS (DADCSS) to improve the sensing adaptiveness and robustness under dynamic environments. It learns environment-invariant features, reduces the domain shift via the inter-domain and intraclass feature alignments, and determines the channel status by an adaptive detector trained by the aligned features. Simulation results demonstrate the effectiveness of our proposed method.
Lusi Li, Jie Li 0024, Yi He 0007, Laura Slayton
CCNC3
2023 IKGN: Intention-aware Knowledge Graph Network for POI Recommendation
abstract
Point-of-Interest (POI) recommendation, pivotal for guiding users to their next interested locale, grapples with the persistent challenge of data sparsity. Whereas knowledge graphs (KGs) have emerged as a favored tool to mitigate the issue, existing KG-based methods tend to overlook two crucial elements: the intention steering users’ location choices and the high-order topological structure within the KG. In this paper, we craft an Intention-aware Knowledge Graph (IKG) that harmonizes users’ visit histories, movement trajectories, and location categories to model user intentions. Building upon IKG, our novel Intention-aware Knowledge Graph Network (IKGN) delves deeper into the POI recommendation by weighing and propagating node embeddings through an attention mechanism, capturing the unique locational intent of each user. A sequential model like GRU is then employed to ensure a comprehensive representation of users’ short- and long-term location preferences. An empirical study on two real-world datasets validates the effectiveness of our proposed IKGN, with it markedly outshining seven benchmark rival models in both Recall and NDCG metrics. The code of IKGN is available at https://github.com/Jungle123456/IKGN.
Chenyang Bu, Bingbing Dong, Shengwei Ji, Yi He 0007, Xindong Wu 0001
ICDM5
2023 Towards Utilitarian Online Learning - A Review of Online Algorithms in Open Feature Space
abstract
Human intelligence comes from the capability to describe and make sense of the world surrounding us, often in a lifelong manner. Online Learning (OL) allows a model to simulate this capability, which involves processing data in sequence, making predictions, and learning from predictive errors. However, traditional OL assumes a fixed set of features to describe data, which can be restrictive. In reality, new features may emerge and old features may vanish or become obsolete, leading to an open feature space. This dynamism can be caused by more advanced or outdated technology for sensing the world, or it can be a natural process of evolution. This paper reviews recent breakthroughs that strived to enable OL in open feature spaces, referred to as Utilitarian Online Learning (UOL). We taxonomize existing UOL models into three categories, analyze their pros and cons, and discuss their application scenarios. We also benchmark the performance of representative UOL models, highlighting open problems, challenges, and potential future directions of this emerging topic.
Yi He 0007, Christian Schreckenberger, Heiner Stuckenschmidt, Xindong Wu 0001
IJCAI1
2023 MMA: Multi-Metric-Autoencoder for Analyzing High-Dimensional and Incomplete Data
Cheng Liang 0003, Di Wu 0056, Yi He 0007, Teng Huang 0001, Zhong Chen 0003, Xin Luo 0001
ECML/PKDD (5)3
2023 A Quantitative Risk Assessment Model for Distribution Cyber-Physical System Under Cyberattack
abstract
An accurate and comprehensive risk assessment in a distribution cyber-physical system (DCPS) is essential to ensure its smooth operation, effective control, and exposure of hidden dangers and security implications. Unfortunately, prior risk assessment methods are mostly limited in two aspects. First, the current studies do not respect the complex network topology and intertwined devicewise dependencies, thereby failing to delineate and model the risk propagation mechanism in DCPS accurately. Second, the prior work tends to focus on gauging the risks underlying physical systems independently, overlooking the quantification of risk impact on physical systems incurred by the cyberattack. To fill the gap, in this article, we propose a unified risk assessment model that gauges the comprehensive security implication of DCPS in a Bayesian regime, in which the three key components, namely, the prior probability, posterior probability, and minimum load-loss ratio, are calculated via the cumulative densities, epidemic model, and optimal load shedding, respectively. We benchmark our proposed model on a public testbed, IEEE 39-bus system, with extensive experiments. The results substantiate the viability and effectiveness of our model and suggest that the vulnerability of DCPS is positively correlated with the three components in our Bayesian modeling. We hope this finding can shed some light on future research by providing a plausible apparatus for measuring the security implications of physical systems in DCPS under cyberattacks.
Song Deng, Jiantang Zhang, Di Wu 0056, Yi He 0007, Xiangpeng Xie 0001, Xindong Wu 0001
IEEE Trans. Ind. Informatics4
2023 Privacy-Preserving Multi-Granular Federated Neural Architecture Search - A General Framework
abstract
Jointly learning from multiple datasets can help building versatile intelligent systems yet may give rise to serious concerns of data privacy and model selection. Specifically, on the one hand, these datasets can be distributed at various local clients, who may not be willing or do not ought to share data with each other. On the other hand, it is unrealistic to choose a model architecture that can well suit the disparate patterns and distributions carried by the various datasets in a priori. Whereas many works in federated learning [1] and neural architecture search [2] have been proposed to address one of the two concerns, very few have attempted the both. To close the gap, in this paper we deliver a framework, termedMulti-Granular Federated Neural Architecture Search(MGFNAS), to enable the automation of model architecture search in a federated and thus privacy-preserved setting. We argue that our MGFNAS framework is general in the sense that it does not impose any restriction on the search space or strategy, such that most existing neural architecture search techniques can be readily implemented in. The main idea of our framework is to search the optimal neural network architecture in two levels of granularity, enabling the neural-operator-basedmicro-levelsearch and the cell-basedmacro-levelsearch. The main challenge of implementing our framework lies in the fact that, due to the decentralized nature, the local architectures searched by multiple clients can differ drastically in order to fit their own datasets, while a general method to form the global model by aggregating the local architectures in both micro and macro levels is missing. To solve the issue, we propose a novel aggregation function, named Network Architecture Probabilistic Aggregation (NAPA). The key idea of our NAPA function is to treat the network architectures as graphs, of which the sub-graph structures being frequently appeared across multiple clients are modeled by probabilistic distributions. At each round, a global model is formed by sampling from those distributions in an exploration-exploitation fashion. Extensive experiments are carried out, and the results substantiate the viability and effectiveness of our proposed framework.
Zijie Pan, Weixuan Tang 0004, Jin Li 0002, Yi He 0007, Zheli Liu
IEEE Trans. Knowl. Data Eng.5
2023 A Lightweight Dynamic Storage Algorithm With Adaptive Encoding for Energy Internet
abstract
Reliable data storage is crucial to the production, transmission, transaction, consumption, and analysis of an Energy Internet (EI). Whereas mainstream distributed data storage seems to be a plausible solution, the existing methods suffer from a tradeoff between the storage overhead (incurred by the replicas of data encodings for lossless recovery) and the communication latency (due to the spiking network traffic resulting from massive queries of data replicas across devices). To balance this tradeoff, we propose aLightweight Dynamic Storage Algorithm based on Adaptive Encoding(LDSA-AE) approach for EI data storage. Our key idea is to classify the data into active and inactive categories, where the active data are most likely to be accessed and thus corrupted in high frequencies. As such, wherever the active data are housed, the replicas of them can be proactively allocated into a set of nearby devices. The main challenges are to realize the classification in real-time and to tailor encoding methods for the active and inactive separately in correspondence to their own characteristics. To overcome these, our LDSA-AE 1) proposes a novel density-based clustering algorithm to tackle performance data classification in an online and unsupervised fashion and 2) leverages Minimum Density RAID-6 (MDR) code and Cauchy Reed-Solomon (CRS) code for active and inactive data encodings, respectively, striving to ensure data storage with low overhead, low latency, high reliability, and high throughput at once. A theoretical analysis substantiates the viability and effectiveness of our proposed LDSA-AE approach. We also prototype our LDSA-AE on a real-world server testbed, and the empirical study suggests the superiority of our approach over the state-of-the-art distributed storage schemes for EI in terms of storage overhead, repair throughput, and reliability.
Song Deng, Di Wu 0056, Dong Yue 0001, Xiong Fu, Yi He 0007
IEEE Trans. Serv. Comput.6
2023 A Double-Space and Double-Norm Ensembled Latent Factor Model for Highly Accurate Web Service QoS Prediction
abstract
Quality-of-Service (QoS), which describes the non-functional characteristics of Web service, is of great significance in service selection. Since users cannot invoke all services to obtain the corresponding QoS data, QoS prediction becomes a hot yet thorny issue. To date, a latent factor analysis (LFA)-based QoS predictor is one of the most successful and popular approaches to address this issue. However, current LFA-based QoS predictors are mostly modeled on inner product space with anL2-norm-orientedLossfunction only. They cannot comprehensively represent the characteristics of target QoS data to make accurate predictions because inner product space andL2-norm have their respective limitations. To address this issue, this study proposes a Double-space and Double-norm Ensembled Latent Factor (D2E-LF) model. Its main idea is three-fold: 1) Double-space—inner product space and distance space are employed to model two kinds of LFA-based QoS predictors, respectively, 2) Double-norm—both of these two predictors adopt anL1-and-L2-norm-orientedLossfunction, and 3) Ensembled—building an ensemble of these two predictors by a weighting strategy. By doing so, D2E-LF integrates multi-merits originating from inner product space, distance space,L1-norm, andL2-norm, making it achieve highly accurate QoS prediction. Experiments on two real-world QoS datasets demonstrate that D2E-LF has significantly higher prediction accuracy than state-of-the-art models.
Di Wu 0056, Peng Zhang 0125, Yi He 0007, Xin Luo 0001
IEEE Trans. Serv. Comput.3
2022 Toward Auto-Learning Hyperparameters for Deep Learning-Based Recommender Systems
Di Wu 0056, Mingsheng Shang 0001, Yi He 0007
DASFAA (2)4
2022 Camouflaged Poisoning Attack on Graph Neural Networks
abstract
Graph neural networks (GNNs) have enabled the automation of many web applications that entail node classification on graphs, such as scam detection in social media and event prediction in service networks. Nevertheless, recent studies revealed that the GNNs are vulnerable to adversarial attacks, where feeding GNNs with poisoned data at training time can lead them to yield catastrophically devastative test accuracy. This finding heats up the frontier of attacks and defenses against GNNs. However, the prior studies mainly posit that the adversaries can enjoy free access to manipulate the original graph, while obtaining such access could be too costly in practice. To fill this gap, we propose a novel attacking paradigm, named Generative Adversarial Fake Node Camouflaging (GAFNC), with its crux lying in crafting a set of fake nodes in a generative-adversarial regime. These nodes carry camouflaged malicious features and can poison the victim GNN by passing their malicious messages to the original graph via learned topological structures, such that they 1) maximize the devastation of classification accuracy (i.e., global attack) or 2) enforce the victim GNN to misclassify a targeted node set into prescribed classes (i.e., target attack). We benchmark our experiments on four real-world graph datasets, and the results substantiate the viability, effectiveness, and stealthiness of our proposed poisoning attack approach. Code is released in github.com/chao92/GAFNC.
Chao Jiang 0002, Yi He 0007, Richard Chapman 0001, Hongyi Wu
ICMR2
2022 Online Deep Learning from Doubly-Streaming Data
abstract
This paper investigates a new online learning problem with doubly-streaming data, where the data streams are described by feature spaces that constantly evolve, with new features emerging and old features fading away. A plausible idea to deal with such data streams is to establish a relationship between the old and new feature spaces, so that an online learner can leverage the knowledge learned from the old features to better the learning performance on the new features. Unfortunately, this idea does not scale up to high-dimensional multimedia data with complex feature interplay, which suffers a tradeoff between onlineness, which biases shallow learners, and expressiveness, which requires deep models. Motivated by this, we propose a novel OLD3S paradigm, where a shared latent subspace is discovered to summarize information from the old and new feature spaces, building an intermediate feature mapping relationship. A key trait of OLD3S is to treat the model capacity as a learnable semantics, aiming to yield optimal model depth and parameters jointly in accordance with the complexity and non-linearity of the input data streams in an online fashion. Both theoretical analysis and empirical studies substantiate the viability and effectiveness of our proposed approach. The code is available online at https://github.com/X1aoLian/OLD3S.
Heng Lian 0001, John Scovil Atwood, Bojian Hou, Jian Wu 0006, Yi He 0007
ACM Multimedia5
2022 A Data-Characteristic-Aware Latent Factor Model for Web Services QoS Prediction
abstract
How to accurately predict unknown quality-of-service (QoS) data based on observed ones is a hot yet thorny issue in Web service-related applications. Recently, a latent factor (LF) model has shown its efficiency in addressing this issue owing to its high accuracy and scalability. An LF model can be improved by identifying user and service neighborhoods based on user and service geographical information. However, such information can be difficult to acquire in most applications with the considerations of information security, identity privacy, and commercial interests in a real system. Besides, the existing LF model-based QoS predictors mostly ignore the reliability of given QoS data where noises commonly exist to cause accuracy loss. To address the above issues, this paper proposes a data-characteristic-aware latent factor (DCALF) model to implement highly accurate QoS predictions, where ‘data-characteristic-aware’ indicates that it can appropriately implement QoS prediction according to the characteristics of given QoS data. Its main idea is two-fold: a) it detects the neighborhoods and noises of users and services based on the dense LFs extracted from the original sparse QoS data, b) it incorporates a density peaks-based clustering method into its modeling process for achieving the simultaneous detections of both neighborhoods and noises of QoS data. With such designs, it precisely represents the given QoS data in spite of their sparsity, thereby achieving highly accurate predictions for unknown ones. Experimental results on two QoS datasets generated by real-world Web services demonstrate that the proposed DCALF model outperforms state-of-the-art QoS predictors, making it highly competitive in addressing the issue of Web service selection and recommendation.
Di Wu 0056, Xin Luo 0001, Mingsheng Shang 0001, Yi He 0007, Guoyin Wang 0001, Xindong Wu 0001
IEEE Trans. Knowl. Data Eng.4
2022 A Posterior-Neighborhood-Regularized Latent Factor Model for Highly Accurate Web Service QoS Prediction
abstract
Neighborhood regularization is highly important for a latent factor (LF)-based Quality-of-Service (QoS)-predictor since similar users usually experience similar QoS when invoking similar services. Current neighborhood-regularized LF models rely prior information on neighborhood obtained from common raw QoS data or geographical information. The former suffers from low prediction accuracy due to the difficulty of constructing the neighborhood based on incomplete QoS data, while the latter requires additional geographical information that is usually difficult to collect considering information security, identity privacy, and commercial interests in real-world scenarios. To address the above issues, this work proposes a posterior-neighborhood-regularized LF (PLF) model for QoS prediction. The main idea is to decompose the LF analysis process into three phases: a) primal LF extraction, where the LFs are extracted to represent involved users/services based on known QoS data, b) posterior-neighborhood construction, where the neighborhood of each user/service is achieved based on similarities between their primal LF vectors, and c) posterior-neighborhood-regularized LF analysis, where the objective function is regularized by both the posterior-neighborhood of users/services and$L_{2}$L2-norm of desired LFs. Experimental results from large scale QoS datasets demonstrate that PLF outperforms state-of-the-art models in terms of both accuracy and efficiency.
Di Wu 0056, Qiang He 0001, Xin Luo 0001, Mingsheng Shang 0001, Yi He 0007, Guoyin Wang 0001
IEEE Trans. Serv. Comput.5
2022 A Latent Factor Analysis-Based Approach to Online Sparse Streaming Feature Selection
abstract
Online streaming feature selection (OSFS) has attracted extensive attention during the past decades. Current approaches commonly assume that the feature space of fixed data instances dynamically increases without any missing data. However, this assumption does not always hold in many real applications. Motivated by this observation, this study aims to implement online feature selection from sparse streaming features, i.e., features flow in one by one with missing data as instance count remains fixed. To do so, this study proposes a latent-factor-analysis-based online sparse-streaming-feature selection algorithm (LOSSA). Its main idea is to apply latent factor analysis to pre-estimate missing data in sparse streaming features before conducting feature selection, thereby addressing the missing data issue effectively and efficiently. Theoretical and empirical studies indicate that LOSSA can significantly improve the quality of OSFS when missing data are encountered in target instances.
Di Wu 0056, Yi He 0007, Xin Luo 0001, MengChu Zhou
IEEE Trans. Syst. Man Cybern. Syst.2
2021 Online Learning in Variable Feature Spaces under Incomplete Supervision
abstract
This paper explores a new online learning problem where the input sequence lives in an over-time varying feature space and the ground-truth label of any input point is given only occasionally, making online learners less restrictive and more applicable. The crux in this setting lies in how to exploit the very limited labels to efficiently update the online learners. Plausible ideas such as propagating labels from labeled points to their neighbors through uncovering the point-wise geometric relations face two challenges: (1) distance measurement fails to work as different points may be described by disparate sets of features and (2) storing the geometric shape, which is formed by all arrived points, is unrealistic in an online setting. To address these challenges, we first construct a universal feature space that accumulates all observed features, making distance measurement feasible. Then, we use manifolds to represent the geometric shapes and approximate them in a sparse means, making manifolds computational and memory tractable in online learning. We frame these two building blocks into a regularized risk minimization algorithm. Theoretical analysis and empirical evidence substantiate the viability and effectiveness of our proposal.
Yi He 0007, Xu Yuan 0001, Sheng Chen 0008, Xindong Wu 0001
AAAI1
2021 Online Learning in Variable Feature Spaces with Mixed Data
abstract
This paper explores a new online learning problem where the data streams are generated from an over-time varying feature space, in which the random variables are of mixed data types including Boolean, ordinal, and continuous. The crux of this setting lies in how to establish the relationship among features, such that the learner can enjoy 1) reconstructed information of the missed-out old features and 2) a jump-start of learning new features with educated weight initialization. Unfortunately, existing methods mainly assume a linear mapping relationship among features or that the multivariate joint distribution could be modeled as Gaussians, limiting their applicability to the mixed data streams. To fill the gap, we in this paper propose to model the complex joint distribution underlying mixed data with Gaussian copula, where the observed features with arbitrary marginals are mapped onto a latent normal space. The feature correlation is approximated in the latent space through an online EM process. Two base learners trained on the observed and latent features are ensembled to expedite convergence, thereby minimizing prediction risk in an online setting. Theoretical and empirical studies substantiate the effectiveness of our proposed approach. Code is released in https://github.com/xiexvying/OVFM.
Yi He 0007, Jiaxian Dong, Bo-Jian Hou, Yu Wang 0017, Fei Wang 0001
ICDM1
2021 Interpretable Minority Synthesis for Imbalanced Classification
abstract
This paper proposes a novel oversampling approach that strives to balance the class priors with a considerably imbalanced data distribution of high dimensionality. The crux of our approach lies in learning interpretable latent representations that can model the synthetic mechanism of the minority samples by using a generative adversarial network(GAN). A Bayesian regularizer is imposed to guide the GAN to extract a set of salient features that are either disentangled or intensionally entangled, with their interplay controlled by a prescribed structure, defined with human-in-the-loop. As such, our GAN enjoys an improved sample complexity, being able to synthesize high-quality minority samples even if the sizes of minority classes are extremely small during training. Empirical studies substantiate that our approach can empower simple classifiers to achieve superior imbalanced classification performance over the state-of-the-art competitors and is robust across various imbalance settings. Code is released in github.com/fudonglin/IMSIC.
Yi He 0007, Fudong Lin, Xu Yuan 0001, Nian-Feng Tzeng
IJCAI1
2021 Unsupervised Lifelong Learning with Curricula
abstract
Lifelong machine learning (LML) has driven the development of extensive web applications, enabling the learning systems deployed on web servers to deal with a sequence of tasks in an incremental fashion. Such systems can retain knowledge from learned tasks in a knowledge base and seamlessly apply it to improve the future learning. Unfortunately, most existing LML methods require labels in every task, whereas providing persistent human labeling for all future tasks is costly, onerous, error-prone, and hence impractical. Motivated by this situation, we propose a new paradigm named unsupervised lifelong learning with curricula (ULLC), where only one task needs to be labeled for initialization and the system then performs lifelong learning for subsequent tasks in an unsupervised fashion. A main challenge of realizing this paradigm lies in the occurrence of negative knowledge transfer, where partial old knowledge becomes detrimental for learning a given task yet cannot be filtered out by the learner without the help of labels. To overcome this challenge, we draw insights from the learning behaviors of humans. Specifically, when faced with a difficult task that cannot be well tackled by our current knowledge, we usually postpone it and work on some easier tasks first, which allows us to grow our knowledge. Thereafter, once we go back to the postponed task, we are more likely to tackle it well as we are more knowledgeable now. The key idea of ULLC is similar – at any time, a pool of candidate tasks are organized in a curriculum by their distances to the knowledge base. The learner then starts from the closer tasks, accumulates knowledge from learning them, and moves to learn the faraway tasks with a gradually augmented knowledge base. The viability and effectiveness of our proposal are substantiated through extensive empirical studies on both synthetic and real datasets.
Yi He 0007, Sheng Chen 0008, Baijun Wu, Xu Yuan 0001, Xindong Wu 0001
WWW1
2021 Toward Mining Capricious Data Streams: A Generative Approach
abstract
Learning with streaming data has received extensive attention during the past few years. Existing approaches assume that the feature space is fixed or changes by following explicit regularities, limiting their applicability in real-time applications. For example, in a smart healthcare platform, the feature space of the patient data varies when different medical service providers use nonidentical feature sets to describe the patients' symptoms. To fill the gap, we in this article propose a novel learning paradigm, namely, Generative Learning With Streaming Capricious (GLSC) data, which does not make any assumption on the feature space dynamics. In other words, GLSC handles the data streams with a varying feature space, where each arriving data instance can arbitrarily carry new features and/or stop carrying partial old features. Specifically, GLSC trains a learner on a universal feature space that establishes relationships between old and new features, so that the patterns learned in the old feature space can be used in the new feature space. The universal feature space is constructed by leveraging the relatednesses among features. We propose a generative graphical model to model the construction process, and show that learning from the universal feature space can effectively improve the performance with theoretical guarantees. The experimental results demonstrate that GLSC achieves conspicuous performance on both synthetic and real data sets.
Yi He 0007, Baijun Wu, Di Wu 0056, Ege Beyazit, Sheng Chen 0008, Xindong Wu 0001
IEEE Trans. Neural Networks Learn. Syst.1
2021 A Deep Latent Factor Model for High-Dimensional and Sparse Matrices in Recommender Systems
abstract
Recommender systems (RSs) commonly adopt a user-item rating matrix to describe users' preferences on items. With users and items exploding, such a matrix is usually high-dimensional and sparse (HiDS). Recently, the idea of deep learning has been applied to RSs. However, current deep-structured RSs suffer from high computational complexity. Enlightened by the idea of deep forest, this paper proposes a deep latent factor model (DLFM) for building a deep-structured RS on an HiDS matrix efficiently. Its main idea is to construct a deep-structured model by sequentially connecting multiple latent factor (LF) models instead of multilayered neural networks through a nonlinear activation function. Thus, the computational complexity grows linearly with its layer count, which is easy to resolve in practice. The experimental results on four HiDS matrices from industrial RSs demonstrate that when compared with state-of-the-art LF models and deep-structured RSs, DLFM can well balance the prediction accuracy and computational efficiency, which well fits the desire of industrial RSs for fast and right recommendations.
Di Wu 0056, Xin Luo 0001, Mingsheng Shang 0001, Yi He 0007, Guoyin Wang 0001, MengChu Zhou
IEEE Trans. Syst. Man Cybern. Syst.4
2020 On Partial Multi-Task Learning
Yi He 0007, Baijun Wu, Di Wu 0056, Xindong Wu 0001
ECAI1
2020 Online Learning to Accelerate Neural Network Inference with Traveling Classifiers
abstract
Deep neural networks trained on millions of instances can recognize a wide variety of patterns. It is common to use these pre-trained deep networks in applications where the domain specific training data is not readily available. Once a pre-trained network is deployed to such applications, some of the information contained in the network may be irrelevant due to the difference between the training set and the application data distributions. As a result, parts of the neural network become redundant and slow down inference. This redundancy is unknown until the model is deployed and input data is received. Therefore, it can only be identified and avoided in real-time. Existing works on neural network acceleration can not exploit such redundancy during offline training when the domain-specific datasets are unavailable. In this paper, we study online learning to accelerate neural network inference. We propose traveling classifiers that continuously learn from the activations of two consecutive network layers to accelerate inference in real-time. Traveling classifiers model class conditional probabilities to generate early predictions and bypass unnecessary computation of network layers. The classifiers also adaptively switch the layers they learn from by measuring the feature space differences between the activations. This traveling mechanism automatically adjusts the aggressiveness of the acceleration without sacrificing prediction accuracy. We demonstrate the performance of the proposed algorithm on the ImageNet dataset [10] using the state-of-the-art ResNet-50, ResNet-152 [18] and VGG-16 [38] architectures. Experiments demonstrate that our method significantly outperforms baseline approaches.
Ege Beyazit, Yi He 0007, Nian-Feng Tzeng, Xindong Wu 0001
ECAI2
2020 Active Learning with Multi-Granular Graph Auto-Encoder
abstract
Predictive modeling of networked data finds many real-world applications, such as fraud detection in social networks, drug discovery in biomedical networks, paper topic classification in citation networks, and so forth. Although the advanced machine learning approaches can help build reasonably accurate predictive models, their applicability is immensely hindered by the data labeling tasks, which are onerous, time-consuming, and error-prone. In this paper, we propose a novel active learning paradigm for networked data, named topology-and-content-aware (TACA) active learning, aiming to minimize the number of labels while achieving a desirable level of model accuracy. Overall, TACA advances existing works from two aspects: (1) TACA makes no assumption on the network property, whereas most existing works only perform effectively on a locally consistent network in which linked nodes are expected to share the same labels and (2) TACA generates queries without relying on model performance, thereby enjoying robust predictive results even when noises exist in the queried labels. Both theoretical and empirical evidences are presented, substantiating the effectiveness of and optimism our approach.
Yi He 0007, Xu Yuan 0001, Nian-Feng Tzeng, Xindong Wu 0001
ICDM1
2020 Learning with Partial Multi-Outlooks
Yi He 0007, Vijay Raghavan 0001
IJCNN2
2020 Deep Matrix Tri-Factorization: Mining Vertex-wise Interactions in Multi-Space Attributed Graphs
abstract
Mining vertex-wise interactions in graphs helps reveal useful information in real-world applications, such as bioinformatics networks BioGRID and DrugBank and academic networks DBLP and Arxiv. A main challenge in developing a general learning method for this setting is that each vertex may be associated with features from heterogeneous feature spaces, representing very disparate information. Moreover, features could be raw and low-level, leading to sparse representations. Some solutions in this area treat all feature spaces as equally important and concatenate features from heterogeneous feature spaces into a single feature vector. Others harmonize different feature spaces by respecting their relative significance in mining vertex-wise interactions but requiring construct specialized harmonizing function and/or handcrafting expressive features, both of which entail expert knowledge. Motivated by this observation, we propose a new learning paradigm named Deep Matrix Tri-Factorization (DM3F), which draws insights from deep models: (i) DM3F replaces the linear combination with a neural architecture that can learn an arbitrary harmonizing function from data; and (ii) DM3F allows raw feature inputs and automatically extracts high-level feature representations via a layer-by-layer learning mechanism. These two characteristics of DM3F make it accessible for users without expert knowledge. DM3F includes two orthogonal and complementary models, allowing an ensemble mechanism to optimize its performance during both training and predicting. A theoretical analysis of DM3F reveals that it possesses several desirable properties, including that it strictly generalizes matrix factorization models. We demonstrate the performance of DM3F on two real-world datasets.
Yi He 0007, Sheng Chen 0008, Thu Nguyen 0001, Bruce A. Wade, Xindong Wu 0001
SDM1
2019 Online Feature Selection with Capricious Streaming Features: A General Framework
abstract
Online streaming feature selection has received extensive attention in the past few years. Existing approaches have a common assumption that the feature space of the fixed data instances increases dynamically without any missing entry. This assumption, however, does not always hold in many real-world applications. For example, in a credit evaluation system, we cannot collect the complete dynamic features for each person and/or enterprise. Motivated by this observation, this paper aims at conducting online feature selection from capricious streaming features, where features flow in one by one with some random missing entries while the number of data instances remains fixed. To do so, we propose a general framework named GF-CSF. The main idea of GF-CSF is to adopt latent factor analysis to preprocess capricious streaming features for completing their missing entries before conducting feature selection. Both theoretical and experimental analyses indicate that GF-CSF can efficiently improve any existing model of online streaming features selection to achieve online capricious streaming features selection.
Di Wu 0056, Yi He 0007, Xin Luo 0001, Mingsheng Shang 0001, Xindong Wu 0001
IEEE BigData2
2019 Online Learning from Capricious Data Streams: A Generative Approach
abstract
Learning with streaming data has received extensive attention during the past few years. Existing approaches assume the feature space is fixed or changes by following explicit regularities, limiting their applicability in dynamic environments where the data streams are described by an arbitrarily varying feature space. To handle such capricious data streams, we in this paper develop a novel algorithm, named OCDS (Online learning from Capricious Data Streams), which does not make any assumption on feature space dynamics. OCDS trains a learner on a universal feature space that establishes relationships between old and new features, so that the patterns learned in the old feature space can be used in the new feature space. Specifically, the universal feature space is constructed by leveraging the relatednesses among features. We propose a generative graphical model to model the construction process, and show that learning from the universal feature space can effectively improve performance with theoretical analysis. The experimental results demonstrate that OCDS achieves conspicuous performance on synthetic and real datasets.
Yi He 0007, Baijun Wu, Di Wu 0056, Ege Beyazit, Sheng Chen 0008, Xindong Wu 0001
IJCAI1
2019 A Data-Aware Latent Factor Model for Web Service QoS Prediction
Di Wu 0056, Xin Luo 0001, Mingsheng Shang 0001, Yi He 0007, Guoyin Wang 0001, Xindong Wu 0001
PAKDD (1)4
2019 Generating precise error specifications for C: a zero shot learning approach
abstract
In C programs, error specifications, which specify the value range that each function returns to indicate failures, are widely used to check and propagate errors for the sake of reliability and security. Various kinds of C analyzers employ error specifications for different purposes, e.g., to detect error handling bugs, yet a general approach for generating precise specifications is still missing. This limits the applicability of those tools. In this paper, we solve this problem by developing a machine learning-based approach named MLPEx. It generates error specifications by analyzing only the source code, and is thus general. We propose a novel machine learning paradigm based on transfer learning, enabling MLPEx to require only one-time minimal data labeling from us (as the tool developers) and zero manual labeling efforts from users. To improve the accuracy of generated error specifications, MLPEx extracts and exploits project-specific information. We evaluate MLPEx on 10 projects, including 6 libraries and 4 applications. An investigation of 3,443 functions and 17,750 paths reveals that MLPEx generates error specifications with a precision of 91% and a recall of 94%, significantly higher than those of state-of-the-art approaches. To further demonstrate the usefulness of the generated error specifications, we use them to detect 57 bugs in 5 tested projects.
Baijun Wu, John Peter Campora III, Yi He 0007, Alexander Schlecht, Sheng Chen 0008
Proc. ACM Program. Lang.3
2018 Supervised Data Synthesizing and Evolving - A Framework for Real-World Traffic Crash Severity Classification
abstract
Traffic crashes have threatened properties and lives for more than thirty years. Thanks to the recent proliferation of traffic data, the machine learning techniques have been broadly expected to make contributions in the traffic safety community due to their triumphs in many other domains. Among these contributions, the most cited method is to classify traffic crashes in different severities since they have significantly unequal occurrences and costs. However, considering the complexity of transportation system, the traffic data are usually highly imbalanced and lowly separable (HILS), so that few proposed works report satisfactory results. In this paper, we propose a novel framework to deal with the HILS traffic crash data. The framework comprises two parts. In part I, a novel Supervised Data Synthesizing and Evolving algorithm is proposed, which can properly represent the HILS data into a more balanced and separable form without altering the original data distribution. In part II, the details of a customized Multi-Layer Perceptron (MLP) are presented, serving the purpose of learning from the represented data with fast convergence and high accuracy. A real-world traffic crash dataset, as a benchmark, is employed to evaluate the classification performances of our framework and three state-of-the-art imbalanced learning algorithms. The experimental results validate that our framework significantly outperforms the other algorithms. Moreover, the impacts of various parameter settings are studied and discussed
Yi He 0007, Di Wu 0056, Ege Beyazit, Xiaoduan Sun, Xindong Wu 0001
ICTAI1