Yi He

Publications

Conference and journal publications, organized by year.

Showing all 97 publications

Filter by topic
Filter by type
2026

U-Net-Guided Detection and Measurement of Growth Bands in Atlantic Surfclam Shell Images

Sophia Sultan, Heng Lian, Roger Mann, and Yi He

IEEE International Conference on Big Data (IEEE Big Data) · High School Symposium · 2026

Uses U-Net-guided image analysis to detect and measure growth bands in Atlantic surfclam shells, supporting the quantitative analysis of shell imagery.

U-NetShell ImagingGrowth Bands

Augmentation with Reinforcement Transformation for Low-textured Scientific Sketch Image Retrieval

Zhiqiang Wang, Sheri L. Barton, Jennifer H. Johnson, Yi He, and Xingquan Zhu

IEEE International Conference on Big Data (IEEE Big Data) · 2026

Studies augmentation with reinforcement transformation for retrieving low-textured scientific sketch images.

Sketch RetrievalData AugmentationReinforcement Transformation

VTV-FM: Flow Matching through Variational Terminal-Velocity Closure

Haoyang Jiang, Yuheng Li, Di Yang, Yanhai Xiong, Haipeng Chen, and Yi He

Conference on Neural Information Processing Systems (NeurIPS) · 2026

Derives missing terminal velocities by minimizing acceleration energy, enabling second-order flow matching with curved transport paths, consistent optimal-transport pairing, and learned acceleration targets.

Flow MatchingGenerative ModelsOptimal Transport

ColdDDI: Evaluating Knowledge Utilization in Cold-Start Drug-Drug Interaction Prediction

Jiheng Liang, Chen Zhao, Di Wu, Chenyang Bu, Yunpeng Hong, Xingquan Zhu, and Yi He

Conference on Neural Information Processing Systems (NeurIPS) · 2026

Benchmarks cold-start drug–drug interaction prediction with mechanism-stratified splits and evidence interventions to distinguish access to biomedical knowledge from models’ actual use of it.

Drug–Drug InteractionsCold StartKnowledge Utilization

Physics-Refined Spatiotemporal Forecasting on Open-Boundary Hydrologic Graphs

Haoyang Jiang, Zhengui Wang, Shenghan Gao, Y. Joseph Zhang, Xingquan Zhu, and Yi He

IEEE International Conference on Data Mining (ICDM) · 2026Best Paper Award

Combines ghost-node proxies for unobserved boundary forcing with local and global physics refiners to stabilize autoregressive forecasting on open-boundary hydrologic graphs.

Hydrologic ForecastingOpen BoundariesPhysics-Informed Learning

An ensemble-based network architecture search framework for 3D dose prediction in radiotherapy

Yang Yan, Chen Zhao, Yi He, Zhenyi Wang, Hao Gao, and Zhong Chen

IEEE International Conference on Data Mining (ICDM) · 2026

Studies ensemble-based neural architecture search for predicting three-dimensional radiotherapy dose distributions, linking automated model design with a clinical treatment-planning prediction task.

Neural Architecture SearchEnsemble LearningRadiotherapy

Online Decision Trees for Varying Feature Spaces via Gradient-Based Surrogates

Kaibu Feng, Heitor M. Gomes, Xingquan Zhu, and Yi He

SIAM International Conference on Data Mining (SDM) · 2026

Develops online decision trees for varying feature spaces through gradient-based surrogate modeling, addressing prediction when the attributes available in a data stream change.

Decision TreesEvolving FeaturesGradient-Based Surrogates

Online Learning for Wild Streaming Data with Delayed Feedback

Yulin Wang, Yi He, Dianlong You, and Di Wu

ACM SIGKDD Conference on Knowledge Discovery and Data Mining (KDD) · 2026

Combines online Gaussian-copula alignment, uncertainty-calibrated geometric pseudo-labels, and dual-track learning to adapt to heterogeneous, incomplete, drifting data streams when ground-truth feedback arrives late.

Delayed FeedbackConcept DriftGeometric Learning

Implicit Fine-tuning via Context Engineering: A Curriculum Learning Framework for Multimodal Entity Alignment

Yunpeng Hong, Chenyang Bu, Di Wu, Yi He, and Xindong Wu

ACM SIGKDD Conference on Knowledge Discovery and Data Mining (KDD) · 2026

Relates context engineering to fine-tuning and uses curriculum-guided, confidence-adaptive prompts to align multimodal entities progressively from simpler to more complex evidence.

Entity AlignmentContext EngineeringCurriculum Learning

PSQE: A Theoretical-Practical Approach to Pseudo Seed Quality Enhancement for Unsupervised Multimodal Entity Alignment

Yunpeng Hong, Chenyang Bu, Jie Zhang, Yi He, Di Wu, and Xindong Wu

ACM SIGKDD Conference on Knowledge Discovery and Data Mining (KDD) · 2026

Improves unsupervised multimodal entity alignment by refining pseudo-seed precision and balancing their graph coverage through multimodal information, clustering, and resampling.

Entity AlignmentPseudo-LabelsMultimodal Learning

MARLIN: Multi-Agent Reinforcement Learning for Incremental DAG Discovery

Dong Li, Zhengzhang Chen, Xujiang Zhao, Linlin Yu, Zhong Chen, Yi He, Haifeng Chen, and Chen Zhao

AAAI Conference on Artificial Intelligence · 2026

Uses state-specific and state-invariant reinforcement-learning agents, a DAG generation policy, and factored actions to incrementally discover causal graph structures from observational data.

Causal DiscoveryMulti-Agent Reinforcement LearningIncremental Learning

Relink: Constructing Query-Driven Evidence Graph On-the-Fly for GraphRAG

Manzong Huang, Chenyang Bu, Yi He, Xingrui Zhuo, and Xindong Wu

AAAI Conference on Artificial Intelligence · 2026

Constructs query-specific evidence graphs by recovering missing relations from source text and selecting relevant facts jointly with existing knowledge-graph edges for question answering.

GraphRAGEvidence GraphsQuestion Answering

Boundary-Consistent Graph Neural Networks for Topological Flux Prediction

Haoyang Jiang, Bojian Qu, Xingquan Zhu, Jifu Tan, and Yi He

Transactions on Machine Learning Research (TMLR) · 2026

Models unobserved external forcing with ghost-node proxies and coupled boundary–interior fixed-point inference to make graph neural networks more consistent for fluid-network flux prediction.

Flux PredictionBoundary ConditionsGhost Nodes

A Novel Spatiotemporally-Regularized Latent Factorization of Tensors Model for Learning Accurate Dynamics from Temporal Sensing Networks

Di Wu, Chengjun Yu, Yi He, and Xin Luo

IEEE Transactions on Networking (TON) · 2026

A spatiotemporally regularized tensor factorization model represents changing sensing-network dynamics by combining latent tensor representations with spatial and temporal structural constraints.

Tensor FactorizationSpatiotemporal ModelingSensing Networks

Federated Latent Factorization of Tensors for Privacy-Preserving Representation Learning to Large-Scale Dynamic Weighted Directed Network

Di Wu, Shuai Zhong, Yi He, Xin Luo, and Xinbo Gao

IEEE Transactions on Dependable and Secure Computing (TDSC) · 2026

FLFT learns latent representations of dynamic weighted directed networks through data-density-aware federated tensor factorization, combining local bias correction, privacy-preserving filling, and nonlinear interaction modeling.

Federated LearningTensor FactorizationDynamic NetworksPrivacy

Non-Gradient Hash Factor Learning for High-Dimensional and Incomplete Data Representation Learning

Di Wu, Shihui Li, Yi He, Xin Luo, and Xinbo Gao

IEEE Transactions on Pattern Analysis and Machine Intelligence (TPAMI) · 2026

NGHF uses discrete differential evolution to optimize binary latent factors directly, learning representations of high-dimensional incomplete data without the quantization step used by continuous relaxations.

Hash LearningDiscrete OptimizationMissing Data

A Survey on Computational Pathology Foundation Models: Datasets, Adaptation Strategies, and Evaluation Tasks

Dong Li, Guihong Wan, Xintao Wu, Xinyu Wu, Yi He, Zhong Chen, Ajit J. Nirmal, Christine G. Lian, Peter K. Sorger, Yevgeniy R. Semenov, and Chen Zhao

Knowledge and Information Systems (KAIS) · 2026

This survey organizes computational pathology foundation models by pretraining datasets, adaptation strategies, and evaluation tasks, highlighting multimodal integration, domain variability, and gaps in standardized assessment.

Computational PathologyFoundation ModelsSurveyModel Evaluation

Multimetric Autoencoder for Representing High-Dimensional and Incomplete Data

Di Wu, Cheng Liang, Yi He, Yan Qiao, and Xin Luo

IEEE Transactions on Systems, Man, and Cybernetics: Systems (TSMC) · 2026

MMA combines four autoencoders defined by different loss and regularization norms, using adaptive weighting to represent heterogeneous high-dimensional data with missing entries.

AutoencodersMulti-Metric LearningMissing Data

Large Language Model-Assisted Research Question Development in Public Health: A Case Study in the Special Supplemental Nutrition Program for Women, Infants, and Children

Qi Zhang, Bidusha Neupane, Priyanka Patel, Futun N. Alkhalifah, Yi He, and Leslie Hodges

Public Health Nutrition (PHN) · 2026

A controlled study evaluates GPT-4 and a scholarly search plugin for generating WIC policy research questions, comparing literature-informed prompting with prompts without literature context through expert assessment.

Large Language ModelsResearch QuestionsPublic HealthWIC
2025

Link Completion for Dynamic Knowledge Graphs with Entity Drift

Xiulin Zheng, Pei-Pei Li, Yi He, and Xindong Wu

IEEE International Conference on Data Mining (ICDM) · 2025

Investigates link completion in dynamic knowledge graphs whose entities drift over time, focusing on maintaining graph relationships as the represented entities evolve.

Entity DriftKnowledge Graph CompletionDynamic Graphs

LHGEL: Large Heterogeneous Graph Ensemble Learning using Batch View Aggregation

Jiajun Shen, Yufei Jin, Yi He, and Xingquan Zhu

IEEE International Conference on Data Mining (ICDM) · 2025

LHGEL ensembles graph learners trained on sampled batch views, combining residual attention and diversity regularization to learn scalable representations of large heterogeneous graphs.

Heterogeneous GraphsEnsemble LearningBatch SamplingResidual Attention

Topology-aware Neural Flux Prediction Guided by Physics

Haoyang Jiang, Jindong Wang, Xingquan Zhu, and Yi He

International Conference on Machine Learning (ICML) · 2025

PhysicsNFP combines directional difference matrices with physical constraints on graph message passing to preserve high-frequency topological signals when predicting flows on directed networks.

Physics-Informed LearningDirected GraphsFlux Prediction

HGEN: Heterogeneous Graph Ensemble Networks

Jiajun Shen, Yufei Jin, Kaibu Feng, Yi He, and Xingquan Zhu

International Joint Conference on Artificial Intelligence (IJCAI) · 2025

HGEN ensembles heterogeneous-graph neural networks built from meta-paths and random dropping, using residual attention and correlation regularization to balance learner accuracy and representational diversity.

Heterogeneous GraphsEnsemble LearningMeta-PathsDiversity Regularization

How to Mitigate Information Loss in Knowledge Graphs for GraphRAG: Leveraging Triple Context Restoration and Query-Driven Feedback

Manzong Huang, Chenyang Bu, Yi He, and Xindong Wu

International Joint Conference on Artificial Intelligence (IJCAI) · 2025

TCR-QF restores source-text context around knowledge-graph triples and iteratively adds query-relevant missing knowledge, addressing information loss in graph-augmented question answering.

GraphRAGContext RestorationKnowledge GraphsQuestion Answering

A Novel Sparse Active Online Learning Framework for Fast and Accurate Streaming Anomaly Detection Over Data Streams

Zhong Chen, Yi He, Di Wu, Chen Zhao, and Meikang Qiu

International Joint Conference on Artificial Intelligence (IJCAI) · 2025

SAD combines sparse online feature selection with CUR-based active instance selection, using reconstruction residuals to score anomalies in evolving streams under computational and memory constraints.

Streaming Anomaly DetectionActive LearningCUR DecompositionFeature Selection

Query-Driven Multimodal GraphRAG: Dynamic Local Knowledge Graph Construction for Online Reasoning

Chenyang Bu, Guojie Chang, Zihao Chen, CunYuan Dang, Zhize Wu, Yi He, and Xindong Wu

Findings of the Association for Computational Linguistics (ACL Findings) · 2025

This framework constructs query-specific multimodal knowledge graphs through semantic graph patterns, multi-path retrieval, and on-demand information completion to support cross-modal question answering.

Multimodal GraphRAGDynamic Knowledge GraphsRetrieval-Augmented GenerationQuestion Answering

Metric-Agnostic Continual Learning for Sustainable Group Fairness

Heng Lian, Chen Zhao, Zhong Chen, Xingquan Zhu, My T. Thai, and Yi He

AAAI Conference on Artificial Intelligence · 2025

MacFRL reorders continual-learning tasks so a classifier learns from confident tasks before tackling larger distribution shifts, aiming to sustain group fairness with unlabeled data.

Continual LearningGroup FairnessCovariate ShiftTask Ordering

ℓ₁,∞-Mixed Norm Promoted Row Sparsity for Fast Online CUR Decomposition Learning in Varying Feature Spaces

Zhong Chen, Yi He, Di Wu, Wenbin Zhang and Zhiqiang Deng

SIAM International Conference on Data Mining (SDM) · 2025

SOOFS combines mixed-norm row sparsity with online CUR decomposition to select informative instances within sliding windows while learning from data streams with changing feature spaces.

Sparse Online LearningCUR DecompositionEvolving FeaturesInstance Selection

Federated Projection Hamming Hash Latent Factor Learning Model

Shihui Li, Di Wu, Yi He and Xin Luo

International Conference on Computer Supported Cooperative Work in Design (CSCWD) · 2025

Introduces a federated latent factor learning model that combines projection operations and Hamming hashing to learn compact representations from distributed data sources.

Federated LearningBinary HashingLatent Factors

SciTableQA: A Question-Answering Benchmark for Complex Scientific Tables

Kehinde Ajayi, Yi He, Kris SeekFord, Matthew Maisonave, and Jian Wu

International Conference on Theory and Practice of Digital Libraries (TPDL) · 2025

Introduces a benchmark of complex scientific tables with human-verified questions that test language models on cell selection and arithmetic reasoning across irregular table layouts.

Table Question AnsweringScientific DocumentsBenchmarking

Uncertainty-Aware Complex Scientific Table Data Extraction

Kehinde Ajayi, Yi He, and Jian Wu

International Conference on Document Analysis and Recognition (ICDAR) · 2025

Combines table structure recognition, optical character recognition, and conformal prediction to flag uncertain extractions and prioritize human verification of complex scientific tables.

Conformal PredictionTable ExtractionOCRUncertainty Quantification

Multi-Modal Foundation Models for Computational Pathology: A Survey

Dong Li, Guihong Wan, Xintao Wu, Xinyu Wu, Xiaohui Chen, Yi He, Zhong Chen, Peter K. Sorger, and Chen Zhao

Transactions on Machine Learning Research (TMLR) · 2025

Surveys computational pathology foundation models that combine histology images with language, knowledge graphs, or gene expression, organizing their datasets, downstream tasks, training methods, and evaluation challenges.

Computational PathologyMultimodal ModelsFoundation ModelsSurvey

Online Outlier Detection in Open Feature Spaces

Heng Lian, Yi He, Di Wu, Zhong Chen, Xingquan Zhu, and Xindong Wu

IEEE Transactions on Knowledge and Data Engineering (TKDE) · 2025

Learns an incrementally updated latent hypersphere from feature embeddings to detect outliers in data streams whose feature spaces change over time.

Outlier DetectionEvolving FeaturesHypersphere Learning

Online Learning from Mix-typed, Drifted, and Incomplete Streaming Features

Shengda Zhuo, Di Wu, Yi He, Shuqiang Huang, and Xindong Wu

ACM Transactions on Knowledge Discovery from Data (TKDD) · 2025

Uses a copula-based latent representation, adaptive drift-detection windows, and geometric label propagation to learn from mixed-type streaming features when distributions shift and labels are incomplete.

Mixed-Type DataConcept DriftIncomplete SupervisionCopula Models

Robust Low-Rank Latent Feature Analysis for Spatiotemporal Signal Recovery

Di Wu, Zechao Li, Zhikai Yu, Yi He, and Xin Luo

IEEE Transactions on Neural Networks and Learning Systems (TNNLS) · 2025

Combines low-rank latent feature modeling, spatiotemporal regularization, and an outlier-resistant loss to recover missing sensor measurements from noisy wireless sensing networks.

Spatiotemporal RecoveryRobust FactorizationSensor Networks
2024

ℓ₁,₂-Norm and CUR Decomposition based Sparse Online Active Learning for Data Streams with Streaming Features

Zhong Chen, Yi He, Di Wu, Liudong Zuo, Keren Li, Wenbin Zhang, and Zhiqiang Deng

IEEE International Conference on Big Data (IEEE Big Data) · 2024

Uses mixed-norm online CUR decomposition to jointly select informative instances and stable features for sparse active learning in streams with evolving feature spaces.

Online Active LearningFeature SelectionCUR DecompositionMixed-Norm Sparsity

HGDL: Heterogeneous Graph Label Distribution Learning

Yufei Jin, Heng Lian, Yi He, and Xingquan Zhu

Conference on Neural Information Processing Systems (NeurIPS) · 2024

Combines meta-path topology homogenization with a topology-and-content-aware graph transformer to learn label distributions for nodes in heterogeneous graphs while preserving structural and semantic consistency.

Heterogeneous GraphsLabel Distribution LearningGraph TransformersMeta-Paths

Utilitarian Online Learning from Open-World Soft Sensing

Heng Lian, Yu Huang, Xingquan Zhu, and Yi He

IEEE International Conference on Data Mining (ICDM) · 2024

Learns a universal representation for evolving soft-sensing features and balances prediction risk against abstention costs when previously unknown classes appear in streaming observations.

Open-World LearningSoft SensingSelective PredictionEvolving Features

Fairness-Aware Streaming Feature Selection with Causal Graphs

Leizhen Zhang, Lusi Li, Di Wu, Sheng Chen, and Yi He

IEEE International Conference on Systems, Man, and Cybernetics (SMC) · 2024

Builds causal graphs around prediction labels and protected attributes to select streaming features while balancing predictive accuracy and fairness as features arrive over time.

Causal FairnessStreaming FeaturesFeature Selection

Learning Gradual Typing Performance

Mohammad Wahiduzzaman Khan, Sheng Chen, and Yi He

European Conference on Object-Oriented Programming (ECOOP) · 2024

Learns runtime models from dynamic type-check overheads to predict how type-annotation configurations affect gradual programs and support performance-aware migration between typing configurations.

Gradual TypingPerformance PredictionType Migration

GLDL: Graph Label Distribution Learning

Yufei Jin, Richard Gao, Yi He, and Xingquan Zhu

AAAI Conference on Artificial Intelligence · 2024

Co-trains graph convolutional networks over instance and label relationships to learn node label distributions while jointly modeling dependencies among instances, labels, and their cross-relations.

Label Distribution LearningGraph ConvolutionLabel Correlations

MKG-FENN: A Multimodal Knowledge Graph Fused End-to-end Neural Network for Accurate Drug-Drug Interaction Prediction

Di Wu, Wu Sun, Yi He, Zhong Chen, and Xin Luo

AAAI Conference on Artificial Intelligence · 2024

MKG-FENN predicts drug–drug interaction events by extracting features from four complementary molecular and biomedical knowledge graphs and fusing them through an end-to-end neural network.

Drug–Drug InteractionsMultimodal Knowledge GraphsGraph Neural NetworksFeature Fusion

Robust Sparse Online Learning for Data Streams with Streaming Features

Zhong Chen, Yi He, Di Wu, Huixin Zhan, Victor S. Sheng, and Kun Zhang

SIAM International Conference on Data Mining (SDM) · 2024

RSOL uses a mixed-norm regularizer and closed-form proximal updates to maintain sparse online predictors when features emerge or disappear from a data stream.

Streaming FeaturesSparse LearningMixed-Norm RegularizationProximal Updates

Multi-type load forecasting model based on random forest and density clustering with the influence of noise and load patterns

Song Deng, Xia Dong, Li Tao, Junjie Wang, Yi He, and Dong Yue

Energy · 2024

MLF-RFDC combines density-based noise detection, latent-factor data recovery, and random-forest forecasting to integrate multiple electrical load types while accounting for distinct consumption patterns.

Load ForecastingRandom ForestsDensity ClusteringNoise Recovery

Online Learning from Evolving Feature Spaces with Deep Variational Models

Heng Lian, Di Wu, Bo-Jian Hou, Jian Wu, and Yi He

IEEE Transactions on Knowledge and Data Engineering (TKDE) · 2024

OLD3S learns shared latent representations and adapts model capacity online, transferring knowledge across evolving feature spaces while retaining information learned from previously observed features.

Feature EvolutionVariational ModelsKnowledge TransferAdaptive Model Capacity

A Debiasing Autoencoder for Recommender System

Teng Huang, Cheng Liang, Di Wu, and Yi He

IEEE Transactions on Consumer Electronics (TCE) · 2024

AutoRec++ combines preprocessing and training bias corrections with mixed loss norms, addressing multiple sources of bias and outliers in autoencoder-based recommendation from user behavior data.

Recommender SystemsAutoencodersDebiasingRobust Losses

A Prediction-Sampling-Based Multilayer-Structured Latent Factor Model for Accurate Representation to High-Dimensional and Sparse Data

Di Wu, Xin Luo, Yi He, and MengChu Zhou

IEEE Transactions on Neural Networks and Learning Systems (TNNLS) · 2024

PMLF builds a multilayer latent-factor architecture that progressively supplements sparse matrices with synthetic entries, using prediction sampling and nonlinear activations to constrain the data-generation process.

Matrix CompletionMultilayer ModelsPrediction SamplingSparse Data
2023

IKGN: Intention-aware Knowledge Graph Network for POI Recommendation

Xiaoyu Zhu, Chenyang Bu, Bingbing Dong, Shengwei Ji, Yi He, and Xindong Wu

IEEE International Conference on Data Mining (ICDM) · 2023

IKGN integrates visit histories, trajectories, and location categories into an intention-aware knowledge graph, combining attention-based propagation with sequential modeling for point-of-interest recommendation.

POI RecommendationUser IntentionsKnowledge GraphsSequential Modeling

MMA: Multi-Metric-Autoencoder for Analyzing High-Dimensional and Incomplete Data

Cheng Liang, Di Wu, Yi He, Teng Huang, Zhong Chen, and Xin Luo

European Conference on Machine Learning and Principles and Practice of Knowledge Discovery in Databases (ECML-PKDD) · 2023

MMA combines four autoencoders with different loss and regularization norms through adaptive weighting, learning representations of heterogeneous high-dimensional data with incomplete observations.

AutoencodersMulti-Metric LearningAdaptive EnsemblesIncomplete Data

Towards Utilitarian Online Learning – A Review of Online Algorithms in Open Feature Space

Yi He, Christian Schreckenberger, Heiner Stuckenschmidt, and Xindong Wu

International Joint Conference on Artificial Intelligence (IJCAI) · 2023

This survey organizes online learning in open feature spaces into a unified taxonomy, compares representative algorithms, and identifies application scenarios, limitations, and research directions.

Open Feature SpacesSurveyAlgorithm TaxonomyBenchmarking

Online Semi-Supervised Learning with Mix-Typed Streaming Features

Di Wu, Shengda Zhuo, Yu Wang, Zhong Chen, and Yi He

AAAI Conference on Artificial Intelligence · 2023

OSLMF uses copula-based feature alignment and online neighborhood label propagation to learn from mixed-type, evolving feature spaces when only a subset of arriving instances is labeled.

Semi-Supervised LearningMixed-Type FeaturesCopulasLabel Propagation

Online Random Feature Forests for Learning in Varying Feature Spaces

Christian Schreckenberger, Yi He, Stefan Lüdtke, Christian Bartelt, and Heiner Stuckenschmidt

AAAI Conference on Artificial Intelligence · 2023

ORF3V maintains an interpretable ensemble of feature-specific decision forests, updating and pruning compressed feature representations as variables appear or disappear in online data streams.

Varying Feature SpacesTree EnsemblesInterpretabilityFeature Pruning

A Graph-Incorporated Latent Factor Analysis Model for High-Dimensional and Sparse Data

Di Wu, Yi He, and Xin Luo

IEEE Transactions on Emerging Topics in Computing (TETC) · 2023

GLFA extracts hidden high-order interactions from sparse matrices as a graph and incorporates them into a recurrent latent-factor architecture for missing-entry prediction.

High-Order InteractionsGraph StructureMatrix CompletionRecurrent Models

A Double-Space and Double-Norm Ensembled Latent Factor Model for Highly Accurate Web Service QoS Prediction

Di Wu, Peng Zhang, Yi He, and Xin Luo

IEEE Transactions on Services Computing (TSC) · 2023

D2E-LF ensembles inner-product and distance-based latent-factor predictors with mixed loss norms, capturing complementary characteristics of sparse web-service quality-of-service observations for missing-value prediction.

QoS PredictionMulti-Metric LearningEnsemble ModelsMatrix Completion

A Quantitative Risk Assessment Model for Distribution Cyber-Physical System Under Cyberattack

Song Deng, Jiantang Zhang, Di Wu, Yi He, Xiangpeng Xie, and Xindong Wu

IEEE Transactions on Industrial Informatics (TII) · 2023

Combines Bayesian risk modeling, epidemic-style propagation, and optimal load shedding to assess how cyberattacks and device dependencies affect the security of distribution cyber-physical systems.

Cyber-Physical SystemsRisk AssessmentBayesian ModelingCybersecurity

Privacy-Preserving Multi-Granular Federated Neural Architecture Search – A General Framework

Zijie Pan, Li Hu, Weixuan Tang, Jin Li, Yi He, and Zheli Liu

IEEE Transactions on Knowledge and Data Engineering (TKDE) · 2023

Investigates multi-granular neural architecture search in federated settings, combining model selection across distributed datasets with the requirement to preserve the privacy of client data.

Federated LearningNeural Architecture SearchData Privacy
2022

Online Deep Learning from Doubly-Streaming Data

Heng Lian, John Scovi Atwood, Bo-Jian Hou, Jian Wu, and Yi He

ACM International Conference on Multimedia (ACM MM) · 2022

Learns a shared latent representation across evolving feature spaces while adapting neural network depth and parameters online to accommodate changes in the complexity of streaming data.

Doubly-Streaming DataFeature EvolutionAdaptive DepthDeep Learning

Online Sparse Streaming Feature Selection with Uncertainty

Feilong Chen, Di Wu, Jie Yang, and Yi He

IEEE International Conference on Networking, Sensing and Control (ICNSC) · 2022

Estimates missing streaming features with latent factor analysis, then combines fuzzy logic and neighborhood rough sets to address uncertainty between imputed feature values and class labels.

Streaming Feature SelectionMissing DataUncertaintyRough Sets

AutoRec++: Incorporating Debias Methods Into Autoencoder-Based Recommender System

Cheng Liang, Yi He, Teng Huang, and Di Wu

IEEE International Conference on Ubiquitous Intelligence and Computing (UIC) · 2022

Combines preprocessing and training debiasing strategies within an autoencoder-based recommender to address biases in user behavior data without changing the underlying neural network architecture.

Recommender SystemsAutoencodersDebiasing

OR-AutoRec: An Outlier-Resilient Autoencoder-Based Recommendation Model

Yuanpeng Hu, Xianmin Wang, Cheng Liang, Jing Li, Di Wu, and Yi He

IEEE International Conference on Ubiquitous Intelligence and Computing (UIC) · 2022

Uses a Cauchy-loss autoencoder to model user–item interactions while reducing the influence of outlier ratings and potentially malicious user behavior on the resulting recommendations.

Recommender SystemsAutoencodersRobust LossOutliers

Camouflaged Poisoning Attack on Graph Neural Networks

Chao Jiang, Yi He, Richard Chapman, and Hongyi Wu

ACM International Conference on Multimedia Retrieval (ICMR) · 2022

Generates camouflaged fake nodes and learned connections to poison graph neural networks for global or targeted misclassification without directly modifying the features of original graph nodes.

Graph Neural NetworksPoisoning AttacksFake NodesAdversarial Learning

A Robust Latent Factor Analysis Model for Incomplete Data Recovery in Wireless Sensor Networks

Zhikai Yu, Di Wu, and Yi He

IEEE International Conference on Edge Computing and Communications (EDGE) · 2022

Studies robust latent factor analysis for recovering incomplete wireless sensor network data, using latent representations to estimate missing measurements from the available sensor observations.

Wireless Sensor NetworksMissing DataRobust Estimation

Toward Auto-Learning Hyperparameters for Deep Learning-Based Recommender Systems

Bo Sun, Di Wu, Mingsheng Shang, and Yi He

International Conference on Database Systems for Advanced Applications (DASFAA) · 2022

Integrates differential evolution into recommender training to learn layer-specific learning rates and regularization coefficients automatically, reducing dependence on manually configured searches for these hyperparameters.

Recommender SystemsHyperparameter OptimizationDifferential Evolution

Quantitative combination load forecasting model based on forecasting error optimization

Song Deng, Fulin Chen, Di Wu, Yi He, Hui Ge, and Yuan Ge

Computers & Electrical Engineering (CEE) · 2022

Combines a K-means and least-squares forecasting-error model with gene expression programming to incorporate uncertain load-data errors into quantitative prediction of electricity demand from observed data.

Load ForecastingError ModelingGene Expression Programming

A Latent Factor Analysis-Based Approach to Online Sparse Streaming Feature Selection

Di Wu, Yi He, Xin Luo, and MengChu Zhou

IEEE Transactions on Systems, Man, and Cybernetics: Systems (TSMC) · 2022

Uses latent factor analysis to estimate missing values before selecting features online, enabling feature selection as incomplete feature columns arrive for a fixed set of instances.

Streaming Feature SelectionMissing DataLatent Factor Analysis

A Data-Characteristic-Aware Latent Factor Model for Web Services QoS Prediction

Di Wu, Xin Luo, Mingsheng Shang, Yi He, Guoyin Wang, and Xindong Wu

IEEE Transactions on Knowledge and Data Engineering (TKDE) · 2022

Uses density-peaks clustering over latent user and service representations to identify neighborhoods and noise, adapting quality-of-service prediction to the characteristics of sparse observations from web services.

QoS PredictionDensity-Peaks ClusteringNoise Detection

A Posterior-neighborhood-regularized Latent Factor Model for Highly Accurate Web Service QoS Prediction

Di Wu, Qiang He, Xin Luo, Mingsheng Shang, Yi He, and Guoyin Wang

IEEE Transactions on Services Computing (TSC) · 2022

Builds user and service neighborhoods from initially learned latent factors, then uses those posterior neighborhoods to regularize quality-of-service predictions without requiring additional geographical information.

QoS PredictionNeighborhood RegularizationCollaborative Filtering
2021

An Ensemble Latent Factor Model for Highly Accurate Web Service QoS Prediction

Peng Zhang, Yi He, and Di Wu

IEEE International Conference on Big Knowledge (ICBK) · 2021

Studies an ensemble of latent factor models for predicting missing web-service quality-of-service values, combining multiple predictors to model sparse observations of interactions between users and services.

QoS PredictionEnsemble LearningMatrix Completion

Online Learning in Variable Feature Spaces with Mixed Data

Yi He, Jiaxian Dong, Bo-Jian Hou, Yu Wang, and Fei Wang

IEEE International Conference on Data Mining (ICDM) · 2021

Models dependencies among mixed-type streaming features with a Gaussian copula, allowing an online learner to transfer information across feature spaces as features appear and disappear.

Evolving Feature SpacesMixed-Type DataGaussian Copulas

Interpretable Minority Synthesis for Imbalanced Classification

Yi He, Fudong Lin, Xu Yuan, and Nian-Feng Tzeng

International Joint Conference on Artificial Intelligence (IJCAI) · 2021

Uses a Bayesian-regularized generative adversarial network to synthesize minority-class examples with interpretable feature dependencies specified through human input, supporting classification when labeled data are imbalanced.

Imbalanced ClassificationGenerative Adversarial NetworksInterpretabilityBayesian Regularization

Unsupervised Lifelong Learning with Curricula

Yi He, Sheng Chen, Baijun Wu, Xu Yuan, and Xindong Wu

The Web Conference (WWW) · 2021

Initializes lifelong learning with one labeled task, then orders unlabeled tasks by their distance to accumulated knowledge to reduce negative transfer.

Lifelong LearningUnsupervised LearningCurriculum Learning

Online Learning in Variable Feature Spaces under Incomplete Supervision

Yi He, Xu Yuan, Sheng Chen, and Xindong Wu

AAAI Conference on Artificial Intelligence · 2021

Combines a universal feature space with sparse manifold regularization to update online classifiers when feature sets vary and supervision is available only intermittently.

Variable FeaturesIncomplete SupervisionManifold Regularization

Toward Wiser Funding Allocation with Bayesian Networks

Jing Chen, Yi He, and Vijay Raghavan

Machine Learning for Consumers and Markets (MLCM) · KDD Workshop · 2021

Examines the use of Bayesian networks to inform funding-allocation decisions, representing uncertain dependencies relevant to distributing limited resources among competing alternatives.

Bayesian NetworksFunding AllocationDecision Support

Toward Mining Capricious Data Streams: A Generative Approach

Yi He, Baijun Wu, Di Wu, Ege Beyazit, Sheng Chen, and Xindong Wu

IEEE Transactions on Neural Networks and Learning Systems (TNNLS) · 2021

Uses a generative graph to reconstruct related features in a universal space, allowing online learning from streams whose features can appear or disappear without predefined rules.

Feature EvolutionGenerative GraphsData Streams

A Deep Latent Factor Model for High-Dimensional and Sparse Matrices in Recommender Systems

Di Wu, Xin Luo, Mingsheng Shang, Yi He, Guoyin Wang, and MengChu Zhou

IEEE Transactions on Systems, Man, and Cybernetics: Systems (TSMC) · 2021

Builds a deep recommender by sequentially connecting latent factor models through nonlinear activations, learning from high-dimensional sparse rating matrices with computation that scales with depth.

Recommender SystemsSparse MatricesLatent Factor Models
2020

Active Learning with Multi-Granular Graph Auto-Encoder

Yi He, Xu Yuan, Nian-Feng Tzeng, and Xindong Wu

IEEE International Conference on Data Mining (ICDM) · 2020

Combines node attributes and network topology through a multi-granular graph autoencoder to select informative instances for labeling, with a query strategy designed to tolerate noisy labels.

Active LearningGraph AutoencodersNoisy Labels

On Partial Multi-Task Learning

Yi He, Baijun Wu, Di Wu, and Xindong Wu

European Conference on Artificial Intelligence (ECAI) · 2020

Learns a shared latent feature space across related tasks and uses a generative graph to recover missing representations, enabling predictions from only partially collected task examples.

Partial ObservationsMulti-Task LearningFeature Recovery

Online Learning to Accelerate Neural Network Inference with Traveling Classifiers

Ege Beyazit, Yi He, N.-F. Tzeng, and Xindong Wu

European Conference on Artificial Intelligence (ECAI) · 2020

Trains traveling classifiers online from consecutive neural-network layers to identify redundant computation after deployment and produce earlier predictions as incoming data reveal application-specific patterns.

Adaptive InferenceEarly ExitsNeural Networks

Learning with Partial Multi-Outlooks

Jing Chen, Yi He, and Vijay Raghavan

International Joint Conference on Neural Networks (IJCNN) · 2020

Uses cross-outlook feature relationships to build a shared repository that reconstructs missing representations, enabling prediction when instances are observed through only some heterogeneous views.

Multi-View LearningMissing ViewsFeature Reconstruction

Deep Matrix Tri-Factorization: Mining Vertex-wise Interactions in Multi-Space Attributed Graphs

Yi He, Sheng Chen, Thu Nguyen, Bruce A. Wade, and Xindong Wu

SIAM International Conference on Data Mining (SDM) · 2020

Learns nonlinear combinations of heterogeneous vertex features through deep matrix tri-factorization, extracting higher-level representations to model interactions between vertices in multi-space attributed graphs.

Attributed GraphsMatrix Tri-FactorizationHeterogeneous Features
2019

Online Feature Selection with Capricious Streaming Features: A General Framework

Di Wu, Yi He, Xin Luo, Mingsheng Shang, and Xindong Wu

IEEE International Conference on Big Data (IEEE Big Data) · 2019

Completes missing entries in newly arriving features with latent factor analysis, allowing existing online feature-selection algorithms to process capricious streams containing incomplete feature observations.

Streaming FeaturesMissing DataFeature Selection

Generating Precise Error Specifications for C: A Zero Shot Learning Approach

Baijun Wu, John Peter Campora III, Yi He, Alexander Schlecht, and Sheng Chen

Object-Oriented Programming, Systems, Languages, and Applications (OOPSLA) · 2019

Combines source-code analysis with transfer learning and project-specific features to infer C error-return specifications, requiring no manual labels from users of previously unseen projects.

Program AnalysisZero-Shot LearningError Specifications

Online Learning from Capricious Data Streams: A Generative Approach

Yi He, Baijun Wu, Di Wu, Ege Beyazit, Sheng Chen, and Xindong Wu

International Joint Conference on Artificial Intelligence (IJCAI) · 2019

Constructs a universal feature space with a generative graphical model so online learners can transfer previously learned patterns to new features in arbitrarily changing data streams.

Feature EvolutionGenerative GraphsData Streams

A Data-Aware Latent Factor Model for Web Service QoS Prediction

Di Wu, Xin Luo, Mingsheng Shang, Yi He, Guoyin Wang, and Xindong Wu

Pacific-Asia Conference on Knowledge Discovery and Data Mining (PAKDD) · 2019

Integrates density-peak clustering into a latent factor model to identify service neighborhoods and unreliable quality-of-service measurements, predicting missing values without requiring additional geographic information.

QoS PredictionDensity-Peak ClusteringUnreliable Data
2018

Supervised Data Synthesizing and Evolving – A Framework for Real-World Traffic Crash Severity Classification

Yi He, Di Wu, Ege Beyazit, Xiaoduan Sun, and Xindong Wu

IEEE International Conference on Tools with Artificial Intelligence (ICTAI) · 2018

Studies a supervised data-synthesis and evolution framework for traffic-crash severity classification, connecting generated training examples with predictive models for real-world transportation records.

Traffic SafetyData SynthesisSeverity Classification

Online Streaming Feature Selection via Conditional Independence

Dianlong You, Xindong Wu, Limin Shen, Yi He, Xu Yuan, Zhen Chen, Song Deng, and Chuan Ma

Applied Sciences · 2018

Applies three layers of conditional-independence filtering to approximate a Markov blanket, removing irrelevant and redundant features as candidate variables arrive in a stream.

Streaming Feature SelectionConditional IndependenceMarkov Blankets