Demo Data from SIGIR 2026 / reviewers in the wild / expert
Yanyan Shen
dblp:74/2905
· DBLP profile ↗
169ranked-venue papers
14as first author
92since 2021 · last 2026
—ORCID · conflict
Domains — the database's venue-derived domains; a paper can count in several
Databases, data management, data science and information retrieval · 87 · 5 first-author · 50 since 2021Artificial intelligence and machine learning · 54 · 2 first-author · 28 since 2021Computer networks · 23 · 5 first-author · 12 since 2021Graphics, computer vision, multimedia, augmented reality and games · 20 · 1 first-author · 11 since 2021Applied, interdisciplinary, general and emerging computing · 13 · 7 since 2021Systems, architecture and hardware · 8 · 1 first-authorSecurity and privacy · 2 · 1 first-authorSoftware engineering, systems software and programming languages · 2 · 1 since 2021Human-computer interaction and ubiquitous computing · 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
46 papers |
Graph learning · 34% Efficient and distributed learning · 24% Deep learning architectures and training · 9% | |
| Databases, data mining, and information retrieval
38 papers |
Recommender systems · 29% Machine learning and data management · 17% Data mining · 14% | |
| Computer architecture, parallel and distributed computing, and storage systems
13 papers |
Distributed systems · 25% Cloud and datacenter computing · 24% Memory systems · 14% |
Topics — the 30 heaviest of 201, each with the papers that count most for it
| Topic | Weight | Papers | Last | Evidence papers |
|---|---|---|---|---|
Machine learning › Graph learning
graph neural network |
4.5 | 8 | 2024 | Self-Interpretable Graph Learning with Sufficient and Necessary Explanations · AAAI 2024 Narrow the Input Mismatch in Deep Graph Neural Network Distillation · KDD 2023 Diga: Guided Diffusion Model for Graph Recovery in Anti-Money Laundering · KDD 2023 |
Machine learning › Graph learning › graph neural network
dynamic graph neural network |
2.9 | 4 | 2025 | A Caching-based Framework for Scalable Temporal Graph Neural Network Training · ACM Trans. Database Syst. 2025 ETC: Efficient Training of Temporal Graph Neural Networks over Large-scale Dynamic Graphs · Proc. VLDB Endow. 2024 Zebra: When Temporal Graph Neural Networks Meet Temporal Personalized PageRank · Proc. VLDB Endow. 2023 |
Machine learning › Graph learning
graph neural network training |
2.6 | 3 | 2026 | Efficient GNN Training on Giant Graphs with Collective Batching and Scheduling · Proc. VLDB Endow. 2026 Faster Convergence in Mini-batch Graph Neural Networks Training with Pseudo Full Neighborhood Compensation · Proc. VLDB Endow. 2025 SIMPLE: Efficient Temporal Graph Neural Network Training at Scale with Dynamic Data Placement · Proc. ACM Manag. Data 2024 |
Machine learning › Graph learning › graph neural network › scalable graph neural network
scalable graph neural network training |
2.0 | 3 | 2024 | SIMPLE: Efficient Temporal Graph Neural Network Training at Scale with Dynamic Data Placement · Proc. ACM Manag. Data 2024 Sancus: Staleness-Aware Communication-Avoiding Full-Graph Decentralized Training in Large-Scale Graph Neural Networks (Extended Abstract) · IJCAI 2023 Feature-Oriented Sampling for Fast and Scalable GNN Training · ICDM 2022 |
Recommender systems
sequential recommendation |
1.9 | 3 | 2024 | A Framework for Elastic Adaptation of User Multiple Intents in Sequential Recommendation · IEEE Trans. Knowl. Data Eng. 2024 Incremental Learning for Multi-Interest Sequential Recommendation · ICDE 2023 Intent Preference Decoupling for User Representation on Online Recommender System · IJCAI 2020 |
Machine learning › Trustworthy machine learning
interpretability |
1.8 | 3 | 2024 | Self-Interpretable Graph Learning with Sufficient and Necessary Explanations · AAAI 2024 Towards Fine-Grained Explainability for Heterogeneous Graph Neural Network · AAAI 2023 Incorporating Interpretability into Latent Factor Models via Fast Influence Analysis · KDD 2019 |
Distributed systems › distributed machine learning
distributed training |
1.7 | 2 | 2026 | Efficient GNN Training on Giant Graphs with Collective Batching and Scheduling · Proc. VLDB Endow. 2026 Sancus: Staleness-Aware Communication-Avoiding Full-Graph Decentralized Training in Large-Scale Graph Neural Networks (Extended Abstract) · IJCAI 2023 |
Machine learning › Efficient and distributed learning
distributed training |
1.6 | 2 | 2025 | From Sancus to Sancusq: staleness and quantization-aware full-graph decentralized training in graph neural networks · VLDB J. 2025 SIMPLE: Efficient Temporal Graph Neural Network Training at Scale with Dynamic Data Placement · Proc. ACM Manag. Data 2024 |
Machine learning › Time series and sequential data › time series analysis
time series forecasting |
1.6 | 2 | 2025 | Proactive Model Adaptation Against Concept Drift for Online Time Series Forecasting · KDD (1) 2025 MASTER: Market-Guided Stock Transformer for Stock Price Forecasting · AAAI 2024 |
Machine learning › Graph learning › graph neural network training
distributed GNN training |
1.6 | 2 | 2026 | Efficient GNN Training on Giant Graphs with Collective Batching and Scheduling · Proc. VLDB Endow. 2026 SANCUS: Staleness-Aware Communication-Avoiding Full-Graph Decentralized Training in Large-Scale Graph Neural Networks · Proc. VLDB Endow. 2022 |
Machine learning › Learning paradigms › incremental learning
concept drift adaptation |
1.5 | 2 | 2025 | Proactive Model Adaptation Against Concept Drift for Online Time Series Forecasting · KDD (1) 2025 DoubleAdapt: A Meta-learning Approach to Incremental Learning for Stock Trend Forecasting · KDD 2023 |
Machine learning › Efficient and distributed learning
embedding cache |
1.5 | 2 | 2025 | A Caching-based Framework for Scalable Temporal Graph Neural Network Training · ACM Trans. Database Syst. 2025 Orca: Scalable Temporal Graph Neural Network Training with Theoretical Guarantees · Proc. ACM Manag. Data 2023 |
Machine learning › Efficient and distributed learning
model compression |
1.5 | 2 | 2025 | Less is More: Unlocking Specialization of Time Series Foundation Models via Structured Pruning · NeurIPS 2025 AutoSrh: An Embedding Dimensionality Search Framework for Tabular Data Prediction · IEEE Trans. Knowl. Data Eng. 2023 |
Machine learning › Efficient and distributed learning › efficient training
training acceleration |
1.5 | 2 | 2025 | A Caching-based Framework for Scalable Temporal Graph Neural Network Training · ACM Trans. Database Syst. 2025 Orca: Scalable Temporal Graph Neural Network Training with Theoretical Guarantees · Proc. ACM Manag. Data 2023 |
Machine learning and data management › deep learning
graph neural network training |
1.4 | 2 | 2024 | Efficient Training of Graph Neural Networks on Large Graphs · Proc. VLDB Endow. 2024 DUCATI: A Dual-Cache Training System for Graph Neural Networks on Giant Graphs with the GPU · Proc. ACM Manag. Data 2023 |
Recommender systems › sequential recommendation
incremental recommendation |
1.4 | 2 | 2024 | A Framework for Elastic Adaptation of User Multiple Intents in Sequential Recommendation · IEEE Trans. Knowl. Data Eng. 2024 Incremental Learning for Multi-Interest Sequential Recommendation · ICDE 2023 |
Query processing and optimization
cardinality estimation |
1.4 | 2 | 2025 | In-Context Adaptation to Concept Drift for Learned Database Operations · ICML 2025 LHist: Towards Learning Multi-dimensional Histogram for Massive Spatial Data · ICDE 2021 |
Recommender systems
click-through rate prediction |
1.3 | 2 | 2023 | RESUS: Warm-up Cold Users via Meta-learning Residual User Preferences in CTR Prediction · ACM Trans. Inf. Syst. 2023 Feature Staleness Aware Incremental Learning for CTR Prediction · IJCAI 2023 |
Data mining
anomaly detection |
1.2 | 2 | 2023 | Diga: Guided Diffusion Model for Graph Recovery in Anti-Money Laundering · KDD 2023 FluxEV: A Fast and Effective Unsupervised Framework for Time-Series Anomaly Detection · WSDM 2021 |
Machine learning and data management
learned database components |
1.1 | 2 | 2025 | In-Context Adaptation to Concept Drift for Learned Database Operations · ICML 2025 NeurDB: an AI-powered autonomous data system · Sci. China Inf. Sci. 2024 |
Indexing and storage engines
learned index |
1.0 | 2 | 2025 | How good are multi-dimensional learned indexes? An experimental survey · VLDB J. 2025 LHist: Towards Learning Multi-dimensional Histogram for Massive Spatial Data · ICDE 2021 |
Computational finance and economics › financial market prediction
stock price prediction |
1.0 | 2 | 2024 | StockMixer: A Simple Yet Strong MLP-Based Architecture for Stock Price Forecasting · AAAI 2024 MASTER: Market-Guided Stock Transformer for Stock Price Forecasting · AAAI 2024 |
Recommender systems
collaborative filtering |
1.0 | 2 | 2022 | LECF: recommendation via learnable edge collaborative filtering · Sci. China Inf. Sci. 2022 Learning Shared Vertex Representation in Heterogeneous Graphs with Convolutional Networks for Recommendation · IJCAI 2019 |
Data integration and cleaning › entity matching
deep entity matching |
0.9 | 2 | 2021 | GNEM: A Generic One-to-Set Neural Entity Matching Framework · WWW 2021 Multi-Context Attention for Entity Matching · WWW 2020 |
Data integration and cleaning
entity matching |
0.9 | 2 | 2021 | GNEM: A Generic One-to-Set Neural Entity Matching Framework · WWW 2021 Multi-Context Attention for Entity Matching · WWW 2020 |
Machine learning › Optimization for machine learning
gradient estimation |
0.9 | 1 | 2025 | Faster Convergence in Mini-batch Graph Neural Networks Training with Pseudo Full Neighborhood Compensation · Proc. VLDB Endow. 2025 |
Machine learning › Graph learning › graph neural network training
mini-batch training |
0.9 | 1 | 2025 | Faster Convergence in Mini-batch Graph Neural Networks Training with Pseudo Full Neighborhood Compensation · Proc. VLDB Endow. 2025 |
Machine learning › Graph learning › graph sampling
neighbor sampling |
0.9 | 1 | 2025 | Faster Convergence in Mini-batch Graph Neural Networks Training with Pseudo Full Neighborhood Compensation · Proc. VLDB Endow. 2025 |
Machine learning › Learning theory
online learning |
0.9 | 1 | 2025 | Proactive Model Adaptation Against Concept Drift for Online Time Series Forecasting · KDD (1) 2025 |
Machine learning › Efficient and distributed learning › large-scale learning
scalable training |
0.9 | 1 | 2025 | A Caching-based Framework for Scalable Temporal Graph Neural Network Training · ACM Trans. Database Syst. 2025 |
Methods — techniques the papers use, named apart from their topics
attention mechanism · 3.1meta-learning · 2.8mini-batch training · 2.6transformer · 2.2greedy search · 2.2scheduling · 2.0auto-tuning · 1.9self-attention · 1.9morton encoding · 1.5historical embedding cache · 1.5bounded embedding staleness · 1.5temporal message passing · 1.5fine-tuning · 1.4capsule network · 1.4graph neural network · 1.1influence functions · 1.0successive convex approximation · 0.9semidefinite relaxation · 0.9
| Year | Publication | Venue | Position |
|---|---|---|---|
| 2026 | Hierarchical learning for IRS-assisted MEC systems with rate-splitting multiple access
Yinyu Wu, Yingchao Jiao, Jinke Ren, Yanyan Shen, Bo Yang 0006, Shuqiang Wang, Dusit Niyato |
Comput. Networks | 5 |
| 2026 | Mil: Cost-guided Minimum Makespan Scheduling for Applications of Multiple LLMs
Jingzhi Fang, Yanyan Shen |
Proc. VLDB Endow. | 2 |
| 2026 | Efficient GNN Training on Giant Graphs with Collective Batching and Scheduling
Xin Zhang 0101, Yanyan Shen, Yingxia Shao, Haoyang Li 0002, Lei Chen 0002 |
Proc. VLDB Endow. | 2 |
| 2025 | NeurDB: On the Design and Implementation of an AI-powered Autonomous Database
Zhanhao Zhao, Shaofeng Cai, Hexiang Pan, Siqi Xiang, Naili Xing, Gang Chen 0001, Beng Chin Ooi, Yanyan Shen, Yuncheng Wu, Meihui Zhang 0001 |
CIDR | 9 |
| 2025 | In-Context Adaptation to Concept Drift for Learned Database OperationsabstractMachine learning has demonstrated transformative potential for database operations, such as query optimization and in-database data analytics. However, dynamic database environments, characterized by frequent updates and evolving data distributions, introduce concept drift, which leads to performance degradation for learned models and limits their practical applicability. Addressing this challenge requires efficient frameworks capable of adapting to shifting concepts while minimizing the overhead of retraining or fine-tuning.
In this paper, we propose FLAIR, an online adaptation framework that introduces a new paradigm called \textit{in-context adaptation} for learned database operations. FLAIR leverages the inherent property of data systems, i.e., immediate availability of execution results for predictions, to enable dynamic context construction. By formalizing adaptation as $f:(\mathbf{x} | \mathcal{C}_t) \to \mathbf{y}$, with $\mathcal{C}_t$ representing a dynamic context memory, FLAIR delivers predictions aligned with the current concept, eliminating the need for runtime parameter optimization. To achieve this, FLAIR integrates two key modules: a Task Featurization Module for encoding task-specific features into standardized representations, and a Dynamic Decision Engine, pre-trained via Bayesian meta-training, to adapt seamlessly using contextual information at runtime. Extensive experiments across key database tasks demonstrate that FLAIR outperforms state-of-the-art baselines, achieving up to $5.2\times$ faster adaptation and reducing error by 22.5\% for cardinality estimation. Jiaqi Zhu 0002, Shaofeng Cai, Yanyan Shen, Gang Chen 0001, Fang Deng, Beng Chin Ooi |
ICML | 3 |
| 2025 | Proactive Model Adaptation Against Concept Drift for Online Time Series ForecastingabstractTime series forecasting always faces the challenge of concept drift, where data distributions evolve over time, leading to a decline in forecast model performance. Existing solutions are based on online learning, which continually organize recent time series observations as new training samples and update model parameters according to the forecasting feedback on recent data. However, they overlook a critical issue: obtaining ground-truth future values of each sample should be delayed until after the forecast horizon. This delay creates a temporal gap between the training samples and the test sample. Our empirical analysis reveals that the gap can introduce concept drift, causing forecast models to adapt to outdated concepts. In this paper, we present Proceed, a novel proactive model adaptation framework for online time series forecasting. Proceed first estimates the concept drift between the recently used training samples and the current test sample. It then employs an adaptation generator to efficiently translate the estimated drift into parameter adjustments, proactively adapting the model to the test sample. To enhance the generalization capability of the framework, Proceed is trained on synthetic diverse concept drifts. Extensive experiments on five real-world datasets across various forecast models demonstrate that Proceed brings more performance improvements than the state-of-the-art online learning methods, significantly facilitating forecast models' resilience against concept drifts. Code is available at https://github.com/SJTU-DMTai/OnlineTSF. Lifan Zhao, Yanyan Shen |
KDD (1) | 2 |
| 2025 | Influence Guided Context Selection for Effective Retrieval-Augmented GenerationabstractRetrieval-Augmented Generation (RAG) addresses large language model (LLM) hallucinations by grounding responses in external knowledge, but its effectiveness is compromised by poor-quality retrieved contexts containing irrelevant or noisy information. While existing approaches attempt to improve performance through context selection based on predefined context quality assessment metrics, they show limited gains over standard RAG. We attribute this limitation to their failure in holistically utilizing available information (query, context list, and generator) for comprehensive quality assessment. Inspired by recent advances in data selection, we reconceptualize context quality assessment as an inference-time data valuation problem and introduce the Contextual Influence Value (CI value). This novel metric quantifies context quality by measuring the performance degradation when removing each context from the list, effectively integrating query-aware relevance, list-aware uniqueness, and generator-aware alignment. Moreover, CI value eliminates complex selection hyperparameter tuning by simply retaining contexts with positive CI values. To address practical challenges of label dependency and computational overhead, we develop a parameterized surrogate model for CI value prediction during inference. The model employs a hierarchical architecture that captures both local query-context relevance and global inter-context interactions, trained through oracle CI value supervision and end-to-end generator feedback. Extensive experiments across 8 NLP tasks and multiple LLMs demonstrate that our context selection method significantly outperforms state-of-the-art baselines, effectively filtering poor-quality contexts while preserving critical information. Code is available at https://github.com/SJTU-DMTai/RAG-CSM. Jiale Deng, Yanyan Shen, Ziyuan Pei, Youmin Chen, Linpeng Huang |
NeurIPS | 2 |
| 2025 | Less is More: Unlocking Specialization of Time Series Foundation Models via Structured PruningabstractScaling laws motivate the development of Time Series Foundation Models (TSFMs) that pre-train vast parameters and achieve remarkable zero-shot forecasting performance. Surprisingly, even after fine-tuning, TSFMs cannot consistently outperform smaller, specialized models trained on full-shot downstream data.
A key question is how to realize effective adaptation of TSFMs for a target forecasting task. Through empirical studies on various TSFMs, the pre-trained models often exhibit inherent sparsity and redundancy in computation, suggesting that TSFMs have learned to activate task-relevant network substructures to accommodate diverse forecasting tasks. To preserve this valuable prior knowledge, we propose a structured pruning method to regularize the subsequent fine-tuning process by focusing it on a more relevant and compact parameter space.
Extensive experiments on seven TSFMs and six benchmarks demonstrate that fine-tuning a smaller, pruned TSFM significantly improves forecasting performance compared to fine-tuning original models. This ``prune-then-finetune'' paradigm often enables TSFMs to achieve state-of-the-art performance and surpass strong specialized baselines.
Source code is made publicly available at \url{https://github.com/SJTU-DMTai/Prune-then-Finetune}. Lifan Zhao, Yanyan Shen, Jiaji Deng |
NeurIPS | 2 |
| 2025 | Rearchitecting the Thread Model of In-Memory Key-Value Stores with μTPSabstractThis paper presents μTPS, a new thread architecture tailored for in-memory key-value stores (KVSs) that operate at tens of millions of operations per second. We show through analysis and demonstration that the widely used run-to-completion thread architecture, which executes monolithic functions from start to finish, often suffers from cache inefficiencies and contention issues. To address this, we revisit the once widely used thread-per-stage (TPS) architecture, but with a fresh perspective – separating cache-resident, contention-free stages and memory-resident, conflict-prone stages into distinct thread pools, and scheduling them with dedicated hardware resources (e.g., CPU cores, cache ways). This novel division enables independent optimization of each stage, significantly improving cache efficiency and mitigating contention. Additionally, μTPS incorporates reconfigurable RPC, resizable caching, and an auto-tuner to enhance its schedulability and performance. We implement two in-memory key-value stores, μTPS-H and μTPS-T, to demonstrate the effectiveness of this approach. Evaluation results show that μTPS achieves higher performance than the run-to-completion counterparts. Youmin Chen, Jiwu Shu, Yanyan Shen, Linpeng Huang, Hong Mei 0001 |
SOSP | 3 |
| 2025 | CLMTR: a generic framework for contrastive multi-modal trajectory representation learning
Anqi Liang, Bin Yao 0002, Jiong Xie, Wenli Zheng, Yanyan Shen, Qiqi Ge |
GeoInformatica | 5 |
| 2025 | ETC: Enhancing Transportation Computation With IRS-Enabled Wireless Powered MEC SystemsabstractThis paper explores the computation enhancement problem in an intelligent reflecting surface (IRS) enabled wireless powered mobile edge computing system serving the intelligent transportation relying on powerful computational support. Initially, in the downlink, the base station (BS) with edge server transmits energy signals to the battery-powered roadside units (RSUs) with computing capabilities, which are grouped into distinct clusters. Subsequently, RSUs leverage the harvested energy for local computing and offloading their tasks to the BS by using a hybrid rate splitting multiple access (RSMA) and time division multiple access (TDMA) strategy. Finally, the computed outcomes from the edge server are transmitted for integration with the computations carried out locally at the RSUs. The objective is to maximize the minimal computation rate of RSU clusters, where the transmit power of the BS and RSUs, the downlink and uplink beamforming of the IRS, the CPU frequency of the RSUs, and the time slot assignment are jointly optimized. To address the bottleneck issues constraining computation rate, this paper proposes an iterative algorithm that combines sequential rank-one constraint relaxation and block coordinate descent methods. Ultimately, the simulation results confirm the effectiveness of the proposed algorithm in enhancing the weakest link constraining the system computation rate when contrasted with the baseline algorithms. Yanyan Shen, Shaobao Li, Shuqiang Wang, Xin-Ping Guan |
IEEE Internet Things J. | 3 |
| 2025 | Latency Minimization for UAV-Enabled Federated Learning: Trajectory Design and Resource AllocationabstractFederated learning (FL) has become a transformative paradigm for distributed machine learning over wireless networks. However, the performance of FL is hindered by the unreliable communication links between resource-constrained Internet of Things (IoT) devices and the central server. To overcome this challenge, we propose a novel framework that employs an unmanned aerial vehicle (UAV) as a mobile server to enhance the FL training process. By capitalizing on the UAV’s mobility, we establish strong line-of-sight connections with IoT devices, thereby enhancing communication reliability and capacity. To maximize training efficiency, we formulate a latency minimization problem that jointly optimizes bandwidth allocation, computing resources, transmit power for both the UAV and IoT devices, and the flight trajectory of the UAV. Subsequently, we analyze the required rounds of the IoT devices training and the UAV aggregation for FL convergence. Based on the convergence constraint, we transform the problem into three subproblems and develop an efficient alternating optimization algorithm to solve this problem. Additionally, we provide a thorough analysis of the algorithm’s convergence and computational complexity. Extensive numerical results demonstrate that the proposed algorithm-based scheme not only surpasses existing benchmark schemes in reducing latency up to 15.29%, but also achieves training efficiency that nearly matches the ideal scenario. Jinke Ren, Huijun Xing, Gui Gui, Yanyan Shen, Shuguang Cui |
IEEE Internet Things J. | 6 |
| 2025 | Apt-Serve: Adaptive Request Scheduling on Hybrid Cache for Scalable LLM Inference ServingabstractLarge language model (LLM) inference serving systems are essential to various LLM-based applications. As demand for LLM services continues to grow, scaling these systems to handle high request rates while meeting latency Service-Level Objectives (SLOs), referred to as effective throughput, becomes critical. However, existing systems often struggle to improve effective throughput, primarily due to a significant decline in Time To First Token (TTFT) SLO attainment. We identify two major causes of this bottleneck: (1) memory-intensive KV cache that limits batch size expansion under GPU memory constraints, and (2) rigid batch composition enforced by the default First-Come-First-Serve scheduling policy. In this paper, we introduce Apt-Serve, a scalable framework designed to enhance effective throughput in LLM inference serving. Apt-Serve features a new hybrid cache scheme that combines KV cache with a memory-efficient hidden cache for reusable input hidden state vectors, allowing large batch sizes and improving request concurrency. Based on the hybrid cache, Apt-Serve employs an adaptive runtime scheduling mechanism that dynamically optimizes batch composition. We formally define the adaptive scheduling optimization problem and propose an efficient algorithm with theoretical guarantees. Extensive evaluations on three real-world datasets and LLMs ranging from 13B to 66B parameters demonstrate that Apt-Serve achieves up to 8.8x improvement in effective throughput compared to the state-of-the-art inference serving systems. Shihong Gao, Xin Zhang 0101, Yanyan Shen, Lei Chen 0002 |
Proc. ACM Manag. Data | 3 |
| 2025 | Faster Convergence in Mini-batch Graph Neural Networks Training with Pseudo Full Neighborhood CompensationabstractGraph Neural Networks (GNNs) have achieved remarkable success in various graph-related tasks. However, training GNNs on large-scale graphs is hindered by the neighbor explosion problem, rendering full-batch training computationally infeasible. Mini-batch training with neighbor sampling is a widely adopted solution, but it introduces gradient estimation errors that slow convergence and reduce model accuracy. In this work, we identify two primary sources of these errors: (1) missing gradient contributions from unsampled target nodes, and (2) inaccuracies in messages computed from sampled nodes. While existing methods largely focus on mitigating the second source, they often overlook the first, resulting in incomplete gradient estimation. To address this gap, we propose the Pseudo Full Neighborhood Compensation (PFNC) framework, which leverages historical information to simultaneously compensate for both errors. PFNC is designed to integrate seamlessly with any neighbor sampling technique and significantly lowers memory demands by maintaining only a partial cache of historical embed-dings and gradients. Theoretical analysis demonstrates that PFNC provides a closer approximation to the ideal gradient, enhancing convergence. Extensive experiments across multiple benchmark datasets confirm that PFNC accelerates convergence and improves generalization across diverse neighbor sampling strategies. Qiqi Zhou 0003, Yanyan Shen, Lei Chen 0002 |
Proc. VLDB Endow. | 2 |
| 2025 | BDHT: Generative AI Enables Causality Analysis for Mild Cognitive ImpairmentabstractEffective connectivity estimation plays a crucial role in understanding the interactions and information flow between different brain regions. However, the functional time series used for estimating effective connectivity is derived from certain software, which may lead to large computing errors because of different parameter settings and degrade the ability to model complex causal relationships between brain regions. In this paper, a brain diffuser with hierarchical transformer (BDHT) is proposed to estimate effective connectivity for mild cognitive impairment (MCI) analysis. To our best knowledge, the proposed brain diffuser is the first generative model to apply diffusion models to the application of generating and analyzing multimodal brain networks. Specifically, the BDHT leverages structural connectivity to guide the reverse processes in an efficient way. It makes the denoising process more reliable and guarantees effective connectivity estimation accuracy. To improve denoising quality, the hierarchical denoising transformer is designed to learn multi-scale features in topological space. By stacking the multi-head attention and graph convolutional network, the graph convolutional transformer (GraphConformer) module is devised to enhance structure-function complementarity and improve the ability in noise estimation. Experimental evaluations of the denoising diffusion model demonstrate its effectiveness in estimating effective connectivity. The proposed model achieves superior performance in terms of accuracy and robustness compared to existing approaches. Moreover, the proposed model can identify altered directional connections and provide a comprehensive understanding of parthenogenesis for MCI treatment.Note to Practitioners—Diagnosing MCI allows for timely intervention and treatment measures to potentially slow down or even halt further cognitive decline. Exploring causal relations between brain regions enables a better understanding of pathogenic mechanisms and the development of effective biomarkers for MCI diagnosis. The current practice heavily relies on the software to analyze MCI causality, leading to large computing errors and degrading MCI analysis performance because of different parameter settings. This work aims to provide a unified framework for the estimation of brain effective connectivity using generative artificial intelligence. Due to their ability to generate high-quality samples, diffusion models have demonstrated remarkable performance in cross-modal medical image synthesis through iterative denoising processes. Our model provides a new insight into how to transform four-dimensional functional magnetic resonance imaging into effective connectivity without relying on software toolkits. The proposed model achieves good disease prediction performance and identifies altered directional connections that may be potential biomarkers for MCI treatment. Our work enables practitioners to develop deep learning model-based medical tools to assist clinicians with disease diagnosis and pathological analysis in an efficient way. Our work can also extend to the intelligently assisted diagnosis of other neurological diseases. Qiankun Zuo, Yanyan Shen, Michael Kwok-Po Ng, Bai Ying Lei, Shuqiang Wang |
IEEE Trans Autom. Sci. Eng. | 3 |
| 2025 | Joint Association, Beamforming, and Resource Allocation for Multi-IRS Enabled MU-MISO Systems With RSMAabstractIntelligent reflecting surface (IRS) and rate-splitting multiple access (RSMA) technologies are at the forefront of enhancing spectrum and energy efficiency in the next generation multi-antenna communication systems. This paper explores a RSMA system with multiple IRSs, and proposes two purpose-driven scheduling schemes, i.e., the exhaustive IRS-aided (EIA) and opportunistic IRS-aided (OIA) schemes. The aim is to optimize the system weighted energy efficiency (EE) under the above two schemes, respectively. Specifically, the Dinkelbach, branch and bound, successive convex approximation, and the semidefinite relaxation methods are exploited within the alternating optimization framework to obtain effective solutions to the considered problems. The numerical findings indicate that the EIA scheme exhibits better performance compared to the OIA scheme in diverse scenarios when considering the weighted EE, and the proposed algorithm demonstrates superior performance in comparison to the baseline algorithms. Huijun Xing, Shuqiang Wang, Yanyan Shen, Bo Yang 0006, Xin-Ping Guan |
IEEE Trans. Mob. Comput. | 6 |
| 2025 | A Caching-based Framework for Scalable Temporal Graph Neural Network TrainingabstractRepresentation learning over dynamic graphs is critical for many real-world applications such as social network services and recommender systems. Temporal graph neural networks (T-GNNs) are powerful representation learning methods and have demonstrated remarkable effectiveness on continuous-time dynamic graphs. However, T-GNNs still suffer from high time complexity, which increases linearly with the number of timestamps and grows exponentially with the model depth, making them not scalable to large dynamic graphs. To address the limitations, we propose Orca , a novel framework that accelerates T-GNN training by caching and reusing intermediate embeddings. We design an optimal caching policy, named MRD , for the uniform cache replacement problem, where embeddings at different intermediate layers have identical dimensions and recomputation costs. MRD not only improves the efficiency of training T-GNNs by maximizing the number of cache hits but also reduces the approximation errors by avoiding keeping and reusing extremely stale embeddings. For the general cache replacement problem, where embeddings at different intermediate layers can have different dimensions and recomputation costs, we solve this NP-hard problem by presenting a novel two-stage framework with approximation guarantees on the achieved benefit of caching. Furthermore, we have developed profound theoretical analyses of the approximation errors introduced by reusing intermediate embeddings, providing a thorough understanding of the impact of our caching and reuse schemes on model outputs. We also offer rigorous convergence guarantees for model training, adding to the reliability and validity of our Orca framework. Extensive experiments have validated that Orca can obtain two orders of magnitude speedup over state-of-the-art T-GNNs while achieving higher precision on various dynamic graphs. Yanyan Shen, Lei Chen 0002, Mingxuan Yuan |
ACM Trans. Database Syst. | 2 |
| 2025 | Age of Information Minimization in UAV-Enabled IoT Networks via Federated Reinforcement LearningabstractThis paper studies the unmanned-aerial-vehicle (UAV)-enabled data collection for Internet-of-things (IoT) networks, in which multiple UAVs are dispatched to collect data over their correspondingly designated areas. In particular, we consider that the UAVs need to collect data in a timely manner. We further consider a practical segmented channel model, in which the air-to-ground wireless channel is assumed to follow Rayleigh or Rician fading when the corresponding line-of-sight (LoS) link is blocked or unblocked, respectively. Under this setup, we minimize the average Age-of-Information (AoI) for the IoT devices, by jointly optimizing the UAV trajectory, the collection scheduling, and the completion time. Since the problem is non-convex, and the dimension of optimization variables varies w.r.t. the completion time that needs to be optimized, the conventional methods are not applicable for efficiently solving the problem. To address this issue, we first propose a deep-reinforcement-learning (DRL) based algorithm to solve this problem in the special case with single UAV, and then exploit federated learning over multiple UAVs to efficiently train the model in a collaborative manner while preserving the data privacy. Numerical results verify that the proposed methods achieve significantly better performance than benchmarks in terms of the AoI, energy consumption, and completion time. Huijun Xing, Yanyan Shen, Jie Xu 0002, Shuguang Cui |
IEEE Trans. Wirel. Commun. | 3 |
| 2025 | How good are multi-dimensional learned indexes? An experimental surveyabstractEfficient indexing is fundamental to managing and analyzing multi-dimensional data. A growing trend is to directly learn the storage layout of multi-dimensional data using simple machine learning models, leading to the concept of Learned Index . Compared to conventional indexing methods that have been used for decades (e.g., k d-tree and R-tree variants), learned indexes have demonstrated empirical advantages in both space and time efficiency on modern architectures. However, there is a lack of comprehensive evaluation across existing multi-dimensional learned indexes under a standardized benchmark, making it challenging to identify the most suitable index for specific data types and query patterns. This gap also hinders the widespread adoption of learned indexes in practical applications. In this paper, we present the first in-depth empirical study to answer the question: how good are multi-dimensional learned indexes? We evaluate ten recently published indexes under a unified experimental framework, which includes standardized implementations, datasets, query workloads, and evaluation metrics. We thoroughly investigate the evaluation results and discuss the findings that may provide insights for future learned index design. Qiyu Liu, Maocheng Li, Yuxiang Zeng, Yanyan Shen, Lei Chen 0002 |
VLDB J. | 4 |
| 2025 | From Sancus to Sancusq: staleness and quantization-aware full-graph decentralized training in graph neural networksabstractAbstract Graph neural networks (GNNs) have emerged due to their success at modeling graph data. Yet, it is challenging for GNNs to efficiently scale to large graphs. Thus, distributed GNNs come into play. To avoid communication caused by expensive data movement between workers, we propose Sancus and its advanced version Sancus"Equation missing", the staleness and quantization-aware communication-avoiding decentralized GNN system. By introducing a set of novel bounded embedding staleness metrics and adaptively skipping broadcasts, Sancus abstracts decentralized GNN processing as sequential matrix multiplication and uses historical embeddings via cache. To further mitigate the communication volume, Sancus"Equation missing" conducts quantization-aware communication on embeddings to reduce the size of broadcast messages. Theoretically, we show bounded approximation errors of embeddings and gradients with a known fastest convergence guarantee. Empirically, we evaluate Sancus and Sancus"Equation missing" with common GNN models via different system setups on large-scale benchmark datasets. Compared to SOTA works, Sancus"Equation missing" can avoid up to $$86\%$$ 86 % communication with $$3.0\times $$ 3.0 × faster throughput on average without accuracy loss. Jingshu Peng, Qiyu Liu, Zhao Chen 0003, Yingxia Shao, Yanyan Shen, Lei Chen 0002, Jiannong Cao 0001 |
VLDB J. | 5 |
| 2024 | Self-Interpretable Graph Learning with Sufficient and Necessary ExplanationsabstractSelf-interpretable graph learning methods provide insights to unveil the black-box nature of GNNs by providing predictions with built-in explanations. However, current works suffer from performance degradation compared to GNNs trained without built-in explanations. We argue the main reason is that they fail to generate explanations satisfying both sufficiency and necessity, and the biased explanations further hurt GNNs' performance. In this work, we propose a novel framework for generating SUfficient aNd NecessarY explanations (SUNNY-GNN for short) that benefit GNNs' predictions. The key idea is to conduct augmentations by structurally perturbing given explanations and employ a contrastive loss to guide the learning of explanations toward sufficiency and necessity directions. SUNNY-GNN introduces two coefficients to generate hard and reliable contrastive samples. We further extend SUNNY-GNN to heterogeneous graphs. Empirical results on various GNNs and real-world graphs show that SUNNY-GNN yields accurate predictions and faithful explanations, outperforming the state-of-the-art methods by improving 3.5% prediction accuracy and 13.1% explainability fidelity on average. Our code and data are available at https://github.com/SJTU-Quant/SUNNY-GNN. Jiale Deng, Yanyan Shen |
AAAI | 2 |
| 2024 | StockMixer: A Simple Yet Strong MLP-Based Architecture for Stock Price ForecastingabstractStock price forecasting is a fundamental yet challenging task in quantitative investment. Various researchers have developed a combination of neural network models (e.g., RNNs, GNNs, Transformers) for capturing complex indicator, temporal and stock correlations of the stock data.While complex architectures are highly expressive, they are often difficult to optimize and the performances are often compromised by the limited stock data. In this paper, we propose a simple MLP-based architecture named StockMixer which is easy to optimize and enjoys strong predictive performance. StockMixer performs indicator mixing, followed by time mixing, and finally stock mixing. Unlike the standard MLP-based mixing, we devise the time mixing to exchange multi-scale time patch information and realize the stock mixing by exploiting stock-to-market and market-to-stock influences explicitly. Extensive experiments on real stock benchmarks demonstrate our proposed StockMixer outperforms various state-of-the-art forecasting methods with a notable margin while reducing memory usage and runtime cost.Code is available at https://github.com/SJTU-Quant/StockMixer. Jinyong Fan, Yanyan Shen |
AAAI | 2 |
| 2024 | MASTER: Market-Guided Stock Transformer for Stock Price ForecastingabstractStock price forecasting has remained an extremely challenging problem for many decades due to the high volatility of the stock market. Recent efforts have been devoted to modeling complex stock correlations toward joint stock price forecasting. Existing works share a common neural architecture that learns temporal patterns from individual stock series and then mixes up temporal representations to establish stock correlations. However, they only consider time-aligned stock correlations stemming from all the input stock features, which suffer from two limitations. First, stock correlations often occur momentarily and in a cross-time manner. Second, the feature effectiveness is dynamic with market variation, which affects both the stock sequential patterns and their correlations. To address the limitations, this paper introduces MASTER, a MArkert-guided Stock TransformER, which models the momentary and cross-time stock correlation and leverages market information for automatic feature selection. MASTER elegantly tackles the complex stock correlation by alternatively engaging in intra-stock and inter-stock information aggregation. Experiments show the superiority of MASTER compared with previous works and visualize the captured realistic stock correlation to provide valuable insights. Tong Li 0017, Zhaoyang Liu 0003, Yanyan Shen |
AAAI | 3 |
| 2024 | Relative Contrastive Learning for Sequential Recommendation with Similarity-based Positive Sample SelectionabstractContrastive Learning (CL) enhances the training of sequential recommendation (SR) models through informative self-supervision signals. Existing methods often rely on data augmentation strategies to create positive samples and promote representation invariance. Some strategies such as item reordering and item substitution may inadvertently alter user intent. Supervised Contrastive Learning (SCL) based methods find an alternative to augmentation-based CL methods by selecting same-target sequences (interaction sequences with the same target item) to form positive samples. However, SCL-based methods suffer from the scarcity of same-target sequences and consequently lack enough signals for contrastive learning. In this work, we propose to use similar sequences (with different target items) as additional positive samples and introduce a Relative Contrastive Learning (RCL) framework for sequential recommendation. RCL comprises a dual-tiered positive sample selection module and a relative contrastive learning module. The former module selects same-target sequences as strong positive samples and selects similar sequences as weak positive samples. The latter module employs a weighted relative contrastive loss, ensuring that each sequence is represented closer to its strong positive samples than its weak positive samples. We apply RCL on two mainstream deep learning-based SR models, and our empirical results reveal that RCL can achieve 4.88% improvement averagely than the state-of-the-art SR methods on five public datasets and one private dataset. Yanyan Shen, Zexi Zhang, Yichun Li |
CIKM | 2 |
| 2024 | Cost-Aware Outdated Facts Correction in the Knowledge Bases
Hao Xin, Lei Chen 0002, Yanyan Shen |
DASFAA (4) | 3 |
| 2024 | PKBC: A Product-Specific Knowledge base Taxonomy Framework
Hao Xin, Lei Chen 0002, Yanyan Shen |
DASFAA (4) | 3 |
| 2024 | StockCL: Selective Contrastive Learning for Stock Trend Forecasting via Learnable Concepts
Zexi Zhang, Lifan Zhao, Yanyan Shen, Bin Yao 0002 |
DASFAA (7) | 3 |
| 2024 | Rethinking Channel Dependence for Multivariate Time Series Forecasting: Learning from Leading IndicatorsabstractRecently, channel-independent methods have achieved state-of-the-art performance in multivariate time series (MTS) forecasting. Despite reducing overfitting risks, these methods miss potential opportunities in utilizing channel dependence for accurate predictions. We argue that there exist locally stationary lead-lag relationships between variates, i.e., some lagged variates may follow the leading indicators within a short time period. Exploiting such channel dependence is beneficial since leading indicators offer advance information that can be used to reduce the forecasting difficulty of the lagged variates. In this paper, we propose a new method named LIFT that first efficiently estimates leading indicators and their leading steps at each time step and then judiciously allows the lagged variates to utilize the advance information from leading indicators. LIFT plays as a plugin that can be seamlessly collaborated with arbitrary time series forecasting methods. Extensive experiments on six real-world datasets demonstrate that LIFT improves the state-of-the-art methods by 5.4% in average forecasting performance. Our code is available at https://github.com/SJTU-Quant/LIFT. Lifan Zhao, Yanyan Shen |
ICLR | 2 |
| 2024 | Intelligent Reflecting Surface Aided Mobile Edge Computing with Rate-Splitting Multiple AccessabstractRecently, intelligent reflecting surface (IRS) has emerged as a promising technology, which can be applied in mobile edge computing (MEC) systems to achieve higher data transmission efficiency and reliability, by providing a reflective channel. Concurrently, rate-splitting multiple access (RSMA), as an innovative technology, is increasingly utilized in MEC systems to enhance data offloading efficiency and facilitate a better integration of computation and communication. In this paper, an IRS enabled MEC system with RSMA under user mobility is considered. Based on this system model, we propose an optimization problem that is aimed at maximizing the system's data transmission rate by jointly optimizing the RSMA power allocation and the IRS phase shift parameters. Although traditional optimization methods can be utilized to solve the considered problem, it is quite time consuming since the optimization methods are often iterative algorithms. To design low complexity algorithm, we propose a deep reinforcement learning (DRL) approach that can efficiently make good decisions quickly after training. Numerical results indicate that, compared to the baseline algorithms, the proposed DRL-based IRS-aided offloading algorithm under RSMA protocol achieves superior system performance. Yinyu Wu, Huijun Xing, Weilin Zang, Shuqiang Wang, Yanyan Shen |
VTC Spring | 6 |
| 2024 | NeurDB: an AI-powered autonomous data system
Beng Chin Ooi, Shaofeng Cai, Gang Chen 0001, Yanyan Shen, Kian-Lee Tan, Yuncheng Wu, Xiaokui Xiao, Naili Xing, Cong Yue, Lingze Zeng, Meihui Zhang 0001, Zhanhao Zhao |
Sci. China Inf. Sci. | 4 |
| 2024 | Joint Trajectory Design and Resource Allocation in UAV-Enabled Heterogeneous MEC SystemsabstractThis article considers a heterogeneous mobile-edge computing (HMEC) system with multiple energy-limited Internet of Things (IoT) devices and an unmanned aerial vehicle (UAV). The UAV can supply energy to all the IoT devices through wireless power transfer. To maximize the utilization of the communication and computation resources, all the IoT devices are divided into two groups, i.e., the active devices and the idle devices. The UAV and the idle devices assist the active devices in executing computing tasks. We formulate an optimization problem that maximizes the minimum task computation data volume among all the active devices by jointly optimizing the UAV trajectory and the communication and computation resource allocation. Since the problem is nonconvex, we decompose the problem into two subproblems: 1) the UAV trajectory design and the computation resource allocation and 2) the time allocation. We utilize a block coordinate descent approach to solve these two subproblems alternately. Simulation results demonstrate that the proposed algorithm can provide an optimized trajectory robust to different initializations. Additionally, compared to the benchmark algorithms, our proposed algorithm shows superior performance in terms of system efficiency and computation data volume. Hao Wang 0240, Huijun Xing, Jinke Ren, Yanyan Shen, Shuguang Cui |
IEEE Internet Things J. | 6 |
| 2024 | Outage Constrained Max-Min Secrecy Rate Optimization for IRS-Aided SWIPT Systems With Artificial NoiseabstractThis study focuses on an intelligent reflecting surface (IRS) enabled simultaneous wireless information and power transfer (SWIPT) system with the coexistence of legitimate users (LUs) and Eavesdroppers (Eves). The main objective is to jointly optimize the transmit beamforming and artificial noise covariance matrix at the access point, the phase shift matrix at the IRS, and the power splitting ratio at the LUs, to maximize the system’s min-secrecy rate. Due to the imperfect channel state information of Eves, an outage rate constraint is contained. The formulated problem is a challenging nonconvex optimization problem since it involves nonconvex objective function and constraints, and the outage rate constraint does not have simple closed form expression. To address this problem, an algorithm based on the alternating optimization method is proposed, which breaks down the nonconvex problem into three subproblems. The algorithm employs several techniques to solve these subproblems. Specifically, the outage rate constraint is approximated using the Bernstein-type inequality. And the Taylor formula, semi-definite relaxation, and successive convex approximation methods are employed to transform the nonconvex subproblems into convex ones. Simulation results demonstrate the effectiveness of the proposed algorithm compared to baseline algorithms under different conditions. Yanyan Shen, Weilin Zang, Bo Yang 0006, Xin-Ping Guan |
IEEE Internet Things J. | 1 |
| 2024 | STile: Searching Hybrid Sparse Formats for Sparse Deep Learning Operators AutomaticallyabstractSparse operators, i.e., operators that take sparse tensors as input, are of great importance in deep learning models. Due to the diverse sparsity patterns in different sparse tensors, it is challenging to optimize sparse operators by seeking an optimal sparse format, i.e., leading to the lowest operator latency. Existing works propose to decompose a sparse tensor into several parts and search for a hybrid of sparse formats to handle diverse sparse patterns. However, they often make a trade-off between search space and search time: their search spaces are limited in some cases, resulting in limited operator running efficiency they can achieve. In this paper, we try to extend the search space in its breadth (by doing flexible sparse tensor transformations) and depth (by enabling multi-level decomposition). We formally define the multi-level sparse format decomposition problem, which is NP-hard, and we propose a framework STile for it. To search efficiently, a greedy algorithm is used, which is guided by a cost model about the latency of computing a sub-task of the original operator after decomposing the sparse tensor. Experiments of two common kinds of sparse operators, SpMM and SDDMM, are conducted on various sparsity patterns, and we achieve 2.1-18.0× speedup against cuSPARSE on SpMMs and 1.5 - 6.9× speedup against DGL on SDDMM. The search time is less than one hour for any tested sparse operator, which can be amortized. Jingzhi Fang, Yanyan Shen, Yue Wang 0012, Lei Chen 0002 |
Proc. ACM Manag. Data | 2 |
| 2024 | SIMPLE: Efficient Temporal Graph Neural Network Training at Scale with Dynamic Data PlacementabstractDynamic graphs are essential in real-world scenarios like social media and e-commerce for tasks such as predicting links and classifying nodes. Temporal Graph Neural Networks (T-GNNs) stand out as a prime solution for managing dynamic graphs, employing temporal message passing to compute node embeddings at specific timestamps. Nonetheless, the high CPU-GPU data loading overhead has become the bottleneck for efficient training of T-GNNs over large-scale dynamic graphs. In this work, we present SIMPLE, a versatile system designed to address the major efficiency bottleneck in training existing T-GNNs on a large scale. It incorporates a dynamic data placement mechanism, which maintains a small buffer space in available GPU memory and dynamically manages its content during T-GNN training. SIMPLE is also empowered by systematic optimizations towards data processing flow. We compare SIMPLE to the state-of-the-art generic T-GNN training system TGL on four large-scale dynamic graphs with different underlying T-GNN models. Extensive experimental results show that SIMPLE effectively cuts down 80.5% ~ 96.8% data loading cost, and accelerates T-GNN training by 1.8× ~ 3.8× (2.6× on average) compared to TGL. Shihong Gao, Xin Zhang 0101, Yanyan Shen, Yingxia Shao, Lei Chen 0002 |
Proc. ACM Manag. Data | 4 |
| 2024 | ETC: Efficient Training of Temporal Graph Neural Networks over Large-scale Dynamic GraphsabstractDynamic graphs play a crucial role in various real-world applications, such as link prediction and node classification on social media and e-commerce platforms. Temporal Graph Neural Networks (T-GNNs) have emerged as a leading approach for handling dynamic graphs, using temporal message passing to compute temporal node embeddings. However, training existing T-GNNs on large-scale dynamic graphs is prohibitively expensive due to the ill-suited batching scheme and significant data access overhead. In this paper, we introduce ETC, a generic framework designed specifically for efficient T-GNN training at scale. ETC incorporates a novel data batching scheme that enables large training batches improving model computation efficiency, while preserving model effectiveness by restricting information loss in each training batch. To reduce data access overhead, ETC employs a three-step data access policy that leverages the data access pattern in T-GNN training, significantly reducing redundant data access volume. Additionally, ETC utilizes an inter-batch pipeline mechanism, decoupling data access from model computation and further reducing data access costs. Extensive experimental results demonstrate the effectiveness of ETC, showcasing its ability to achieve significant training speedups compared to state-of-the-art training frameworks for T-GNNs on real-world dynamic graphs with millions of interactions. ETC provides a training speedup ranging from 1.6X to 62.4X, highlighting its potential for efficient training on large-scale dynamic graphs. Shihong Gao, Yanyan Shen, Yingxia Shao, Lei Chen 0002 |
Proc. VLDB Endow. | 3 |
| 2024 | Efficient Training of Graph Neural Networks on Large GraphsabstractGraph Neural Networks (GNNs) have gained significant popularity for learning representations of graph-structured data. Mainstream GNNs employ the message passing scheme that iteratively propagates information between connected nodes through edges. However, this scheme incurs high training costs, hindering the applicability of GNNs on large graphs. Recently, the database community has extensively researched effective solutions to facilitate efficient GNN training on massive graphs. In this tutorial, we provide a comprehensive overview of the GNN training process based on the graph data lifecycle, covering graph preprocessing, batch generation, data transfer, and model training stages. We discuss recent data management efforts aiming at accelerating individual stages or improving the overall training efficiency. Recognizing the distinct training issues associated with static and dynamic graphs, we first focus on efficient GNN training on static graphs, followed by an exploration of training GNNs on dynamic graphs. Finally, we suggest some potential research directions in this area. We believe this tutorial is valuable for researchers and practitioners to understand the bottleneck of GNN training and the advanced data management techniques to accelerate the training of different GNNs on massive graphs in diverse hardware settings. Yanyan Shen, Lei Chen 0002, Jingzhi Fang, Xin Zhang 0101, Shihong Gao |
Proc. VLDB Endow. | 1 |
| 2024 | Estimating Addiction-Related Brain Connectivity by Prior-Embedding Graph Generative Adversarial NetworksabstractThe study of nicotine addiction mechanism is of great significance in both nicotine withdrawal and brain science. The detection of addiction-related brain connectivity using functional magnetic resonance imaging (fMRI) is a critical step in study of this mechanism. However, it is challenging to accurately estimate addiction-related brain connectivity due to the low-signal-to-noise ratio of fMRI and the issue of small sample size. In this work, a prior-embedding graph generative adversarial network (PG-GAN) is proposed to capture addiction-related brain connectivity accurately. By designing a dual-generator-based scheme, the addiction-related connectivity generator is employed to learn the feature map of addiction connection, while the reconstruction generator is used for sample reconstruction. Moreover, a bidirectional mapping mechanism is designed to maintain the consistency of sample distribution in the latent space so that addiction-related brain connectivity can be estimated more accurately. The proposed model utilizes prior knowledge embeddings to reduce the search space so that the model can better understand the latent distribution for the issue of small sample size. Experimental results demonstrate the effectiveness of the proposed PG-GAN. Changhong Jing, Yanyan Shen, Yi Pan 0001, C. L. Philip Chen, Bai Ying Lei, Shuqiang Wang |
IEEE Trans. Cybern. | 2 |
| 2024 | A Framework for Elastic Adaptation of User Multiple Intents in Sequential RecommendationabstractRecently, substantial research has been conducted on sequential recommendation, with the objective of forecasting the subsequent item by leveraging a user's historical sequence of interacted items. Prior studies employ both capsule networks and self-attention techniques to effectively capture diverse underlying intents within a user's interaction sequence, thereby achieving the most advanced performance in sequential recommendation. However, users could potentially form novel intents from fresh interactions as the lengths of user interaction sequences grow. Consequently, models need to be continually updated or even extended to adeptly encompass these emerging user intents, referred as incremental multi-intent sequential recommendation. In this paper, we propose an effectiveIncremental learning framework for userMulti-intentAdaptation in sequential recommendation called IMA, which augments the traditional fine-tuning strategy with the existing-intents retainer, new-intents detector, and projection-based intents trimmer to adaptively expand the model to accommodate user's new intents and prevent it from forgetting user's existing intents. Furthermore, we upgrade the IMA into anElasticMulti-intentAdaptation (EMA) framework which can elastically remove inactive intents and compress user intent vectors under memory space limit. Extensive experiments on real-world datasets verify the effectiveness of the proposed IMA and EMA on incremental multi-intent sequential recommendation, compared with various baselines. Yanyan Shen |
IEEE Trans. Knowl. Data Eng. | 2 |
| 2023 | Towards Fine-Grained Explainability for Heterogeneous Graph Neural NetworkabstractHeterogeneous graph neural networks (HGNs) are prominent approaches to node classification tasks on heterogeneous graphs. Despite the superior performance, insights about the predictions made from HGNs are obscure to humans. Existing explainability techniques are mainly proposed for GNNs on homogeneous graphs. They focus on highlighting salient graph objects to the predictions whereas the problem of how these objects affect the predictions remains unsolved. Given heterogeneous graphs with complex structures and rich semantics, it is imperative that salient objects can be accompanied with their influence paths to the predictions, unveiling the reasoning process of HGNs. In this paper, we develop xPath, a new framework that provides fine-grained explanations for black-box HGNs specifying a cause node with its influence path to the target node. In xPath, we differentiate the influence of a node on the prediction w.r.t. every individual influence path, and measure the influence by perturbing graph structure via a novel graph rewiring algorithm. Furthermore, we introduce a greedy search algorithm to find the most influential fine-grained explanations efficiently. Empirical results on various HGNs and heterogeneous graphs show that xPath yields faithful explanations efficiently, outperforming the adaptations of advanced GNN explanation approaches. Tong Li 0017, Jiale Deng, Yanyan Shen, Luyu Qiu, Yongxiang Huang, Caleb Chen Cao |
AAAI | 3 |
| 2023 | Human-in-the-Loop Vehicle ReIDabstractVehicle ReID has been an active topic in computer vision, with a substantial number of deep neural models proposed as end-to-end solutions. In this paper, we solve the problem from a new perspective and present an interesting variant called human-in-the-loop vehicle ReID to leverage interactive (and possibly wrong) human feedback signal for performance enhancement. Such human-machine cooperation mode is orthogonal to existing ReID models. To avoid incremental training overhead, we propose an Interaction ReID Network (IRIN) that can directly accept the feedback signal as an input and adjust the embedding of query image in an online fashion. IRIN is offline trained by simulating the human interaction process, with multiple optimization strategies to fully exploit the feedback signal. Experimental results show that even by interacting with flawed feedback generated by non-experts, IRIN still outperforms state-of-the-art ReID models by a considerable margin. If the feedback contains no false positive, IRIN boosts the mAP in Veri776 from 81.6% to 95.2% with only 5 rounds of interaction per query image. Zepeng Li 0002, Dongxiang Zhang, Yanyan Shen, Gang Chen 0001 |
AAAI | 3 |
| 2023 | Rainfall Spatial Interpolation with Graph Neural Networks
Jia Li 0014, Yanyan Shen, Lei Chen 0002, Charles Wang Wai Ng |
DASFAA (4) | 2 |
| 2023 | TE-DyGE: Temporal Evolution-Enhanced Dynamic Graph Embedding Network
Liping Wang 0015, Yanyan Shen, Lei Chen 0002 |
DASFAA (3) | 2 |
| 2023 | HIT - An Effective Approach to Build a Dynamic Financial Knowledge Base
Hao Xin, Yanyan Shen, Lei Chen 0002 |
DASFAA (2) | 3 |
| 2023 | Incremental Learning for Multi-Interest Sequential RecommendationabstractIn recent years, sequential recommendation has been widely researched, which aims to predict the next item of interest based on user’s previously interacted item sequence. Existing works utilize capsule network and self-attention method to explicitly capture multiple underlying interests from a user’s interaction sequence, achieving the state-of-the-art sequential recommendation performance. In practice, the lengths of user interaction sequences are ever-increasing and users might develop new interests from new interactions, and a model should be updated or even expanded continuously to capture the new user interests. We refer to this problem as incremental multi-interest sequential recommendation, which has not yet been well investigated in the existing literature. In this paper, we propose an effective incremental learning framework for multi-interest sequential recommendation called IMSR, which augments the traditional fine-tuning strategy with the existing-interests retainer (EIR), new-interests detector (NID), and projection-based interests trimmer (PIT) to adaptively expand the model to accommodate user’s new interests and prevent it from forgetting user’s existing interests. Extensive experiments on real-world datasets verify the effectiveness of the proposed IMSR on incremental multi-interest sequential recommendation, compared with various baseline approaches. Yanyan Shen |
ICDE | 2 |
| 2023 | Sancus: Staleness-Aware Communication-Avoiding Full-Graph Decentralized Training in Large-Scale Graph Neural Networks (Extended Abstract)abstractGraph neural networks (GNNs) have emerged due to their success at modeling graph data. Yet, it is challenging for GNNs to efficiently scale to large graphs. Thus, distributed GNNs come into play. To avoid communication caused by expensive data movement between workers, we propose SANCUS, a staleness-aware communication-avoiding decentralized GNN system. By introducing a set of novel bounded embedding staleness metrics and adaptively skipping broadcasts, SANCUS abstracts decentralized GNN processing as sequential matrix multiplication and uses historical embeddings via cache. Theoretically, we show bounded approximation errors of embeddings and gradients with convergence guarantee. Empirically, we evaluate SANCUS with common GNN models via different system setups on large-scale benchmark datasets. Compared to SOTA works, SANCUS can avoid up to 74% communication with at least 1:86_ faster throughput on average without accuracy loss. Jingshu Peng, Zhao Chen 0003, Yingxia Shao, Yanyan Shen, Lei Chen 0002, Jiannong Cao 0001 |
IJCAI | 4 |
| 2023 | Feature Staleness Aware Incremental Learning for CTR PredictionabstractClick-through Rate (CTR) prediction in real-world recommender systems often deals with billions of user interactions every day. To improve the training efficiency, it is common to update the CTR prediction model incrementally using the new incremental data and a subset of historical data. However, the feature embeddings of a CTR prediction model often get stale when the corresponding features do not appear in current incremental data. In the next period, the model would have a performance degradation on samples containing stale features, which we call the feature staleness problem. To mitigate this problem, we propose a Feature Staleness Aware Incremental Learning method for CTR prediction (FeSAIL) which adaptively replays samples containing stale features. We first introduce a staleness-aware sampling algorithm (SAS) to sample a fixed number of stale samples with high sampling efficiency. We then introduce a staleness-aware regularization mechanism (SAR) for a fine-grained control of the feature embedding updating. We instantiate FeSAIL with a general deep learning-based CTR prediction model and the experimental results demonstrate FeSAIL outperforms various state-of-the-art methods on four benchmark datasets. The code can be found in https://github.com/cloudcatcher888/FeSAIL. Yanyan Shen, Zibin Zhang, Kangyi Lin |
IJCAI | 2 |
| 2023 | Diga: Guided Diffusion Model for Graph Recovery in Anti-Money LaunderingabstractWith the upsurge of online banking, mobile payment, and virtual currency, new money-laundering crimes easily conceal in the enormous transaction volume. The traditional rule-based methods with large amounts of alerting thresholds are already incapable of handling the fast-changing transaction networks. Recently, the DL models represented by the graph neural networks (GNNs) show the potential to capture money-laundering modes with high accuracy. However, most related works are still far from practical deployment in the industry. Based on our practice at WeBank, there are three major challenges: Firstly, supervised learning is infeasible facing the extraordinarily large-scale but imbalanced data, with hundreds of millions of active accounts but only thousands of anomalies. Secondly, the real-world transactions form a sparse network with millions of isolated user groups, which overflows the expressive ability of current node-level GNNs. Thirdly, the explanation for each suspicious account is mandatory by the government for double check, which conflicts with the black-box nature of most DL models. Therefore, we proposed Diga, the first work to apply the diffusion probabilistic model to a graph anomaly detection problem with three novel techniques: the biased K-hop PageRank, the semi-supervised guided diffusion and the novel weight-sharing GNN layer. The effectiveness and efficiency of Diga are verified via intensive experiments on both industrial and public datasets. Xujia Li, Xueying Mo, Hebing Xiao, Yanyan Shen, Lei Chen 0002 |
KDD | 5 |
| 2023 | DoubleAdapt: A Meta-learning Approach to Incremental Learning for Stock Trend ForecastingabstractStock trend forecasting is a fundamental task of quantitative investment where precise predictions of price trends are indispensable. As an online service, stock data continuously arrive over time. It is practical and efficient to incrementally update the forecast model with the latest data which may reveal some new patterns recurring in the future stock market. However, incremental learning for stock trend forecasting still remains under-explored due to the challenge of distribution shifts (a.k.a. concept drifts). With the stock market dynamically evolving, the distribution of future data can slightly or significantly differ from incremental data, hindering the effectiveness of incremental updates. To address this challenge, we propose DoubleAdapt, an end-to-end framework with two adapters, which can effectively adapt the data and the model to mitigate the effects of distribution shifts. Our key insight is to automatically learn how to adapt stock data into a locally stationary distribution in favor of profitable updates. Complemented by data adaptation, we can confidently adapt the model parameters under mitigated distribution shifts. We cast each incremental learning task as a meta-learning task and automatically optimize the adapters for desirable data adaptation and parameter initialization. Experiments on real-world stock datasets demonstrate that DoubleAdapt achieves state-of-the-art predictive performance and shows considerable efficiency. Lifan Zhao, Shuming Kong, Yanyan Shen |
KDD | 3 |
| 2023 | Narrow the Input Mismatch in Deep Graph Neural Network DistillationabstractGraph neural networks (GNNs) have been widely studied for modeling graph-structured data. Thanks to the over-parameterization and large receptive field of deep GNNs, "deep" is a promising direction to develop GNNs further and has shown some superior performances. However, the over-stacked structures of deep architectures incur high inference cost in deployment. To compress deep GNNs, we can use knowledge distillation (KD) to make shallow student GNNs mimic teacher GNNs. Existing KD methods in graph domain focus on constructing diverse supervision on embedding or prediction produced by student GNNs, but overlook the gap of the receptive field (i.e., input information) between student and teacher, which brings difficulties to KD. We call this gap "input mismatch". To alleviate this problem, we propose a lightweight stochastic extended module to provide an estimation for missing input information for student GNNs. The estimator models the distribution of missing information. Specifically, we model the missing information as an independent distribution from graph level and a conditional distribution from node level (given the condition of observable input). These two estimates are optimized using a Bayesian methodology and combined into a balanced estimate as additional input to student GNNs. To the best of our knowledge, we are the first to address the "input mismatch" problem in deep GNNs distillation. Experiments on extensive benchmarks demonstrate that our method outperforms existing KD methods for GNNs in distillation performance, which confirms that the estimations are reasonable and effective. Qiqi Zhou 0003, Yanyan Shen, Lei Chen 0002 |
KDD | 2 |
| 2023 | Average AoI minimization for data collection in UAV-enabled IoT backscatter communication systems with the finite blocklength regimeabstractThanks to the autonomy of unmanned aerial vehicle (UAV), UAV-enabled data collection in Internet of Things (IoT) networks has become a key application for the next generation communication network. In this paper, we consider a scenario where an UAV is responsible for collecting data from sensor equipments (SEs) one by one and finally carrying the collected data to the computation center for processing. Different from the commonly used assumption that SEs always generate data at the beginning of each time slot, it is assumed that SEs can generate data at any instant during one time slot, which is more practical. Since SEs are energy limited, they upload data to the UAV by adopting backscatter communication technology to reduce energy consumption. Meanwhile, the updated information usually contains a small number of information bits but requires low latency and high reliability, thus the finite blocklength regime in ultra-reliable and low-latency communication is adopted for SEs’ data transmission to the UAV. To keep the freshness of the updated information, a joint resource allocation problem including data collection time allocation, transmission power and trajectory design of the UAV is formulated as an optimization problem to minimize the average age of information (AoI) of all SEs. The formulated problem mixes discrete and continuous variables, which makes it difficult to solve. Thereby, we decompose the optimization problem into data collection time minimization subproblem and UAV trajectory design subproblem, which are solved by the successive convex approximation method, and the backtracking algorithm and the genetic algorithm , respectively. Numerical results show that the backtracking-based algorithm that can obtain the optimal trajectory of the UAV gains the minimal average AoI, and the genetic-based algorithm achieves sub-optimal average AoI with much lower computational complexity. The results also demonstrate that the average AoI of the backtracking-based algorithm and the genetic-based algorithm is reduced by up to 54% and 46% compared with the greedy-based benchmark algorithm, respectively. Yanyan Shen, Weiran Luo, Shuqiang Wang, Xiaoxia Huang 0003 |
Ad Hoc Networks | 1 |
| 2023 | Maximizing Energy Efficiency in UAV-Assisted NOMA-MEC NetworksabstractMobile-edge computing (MEC) is a key technology to enable multitasking and low-latency user experiences for 5G Internet of Things (IoT) devices. The nonorthogonal multiple access (NOMA) technology is used in this context to enable large-scale connectivity and improve spectrum efficiency, with the unmanned aerial vehicle (UAV) serving as both computing units and relays for mobile users (MUs). Energy efficiency (EE) remains challenging given the limited energy available to the UAV and MUs. In this article, a UAV-assisted NOMA–MEC communication network architecture is studied to maximize the EE of the total system by jointly optimizing the user’s communication scheduling, resource allocation, and UAV flight trajectory. Among them, the resource allocation problem can further be divided into the transmit power optimization problem and the task computation allocation problem, whereby the corresponding time slot scheduling is obtained. The objective function is a nonconvex mixed-integer nonlinear fractional programming (MINLFP) problem, which is too complex to solve directly. Therefore, it is decomposed into more manageable subproblems and solved iteratively. Fractional problems are solved using the Dinkelbach method, which transforms their original subproblems into convex forms with methods such as successive convex approximation (SCA). Simulation results demonstrate the convergence of our proposed algorithm and its significant advantage over existing strategies in terms of EE. Zhixin Liu 0001, Junxiao Qi, Yanyan Shen, Kai Ma 0001, Xin-Ping Guan |
IEEE Internet Things J. | 3 |
| 2023 | SSIN: Self-Supervised Learning for Rainfall Spatial InterpolationabstractThe acquisition of accurate rainfall distribution in space is an important task in hydrological analysis and natural disaster pre-warning. However, it is impossible to install rain gauges on every corner. Spatial interpolation is a common way to infer rainfall distribution based on available raingauge data. However, the existing works rely on some unrealistic pre-settings to capture spatial correlations, which limits their performance in real scenarios. To tackle this issue, we propose the SSIN, which is a novel data-driven self-supervised learning framework for rainfall spatial interpolation by mining latent spatial patterns from historical observation data. Inspired by the Cloze task and BERT, we fully consider the characteristics of spatial interpolation and design the SpaFormer model based on the Transformer architecture as the core of SSIN. Our main idea is: by constructing rich self-supervision signals via random masking, SpaFormer can learn informative embeddings for raw data and then adaptively model spatial correlations based on rainfall spatial context. Extensive experiments on two real-world raingauge datasets show that our method outperforms the state-of-the-art solutions. In addition, we take traffic spatial interpolation as another use case to further explore the performance of our method, and SpaFormer achieves the best performance on one large real-world traffic dataset, which further confirms the effectiveness and generality of our method. Jia Li 0014, Yanyan Shen, Lei Chen 0002, Charles Wang Wai Ng |
Proc. ACM Manag. Data | 2 |
| 2023 | Orca: Scalable Temporal Graph Neural Network Training with Theoretical GuaranteesabstractRepresentation learning over dynamic graphs is critical for many real-world applications such as social network services and recommender systems. Temporal graph neural networks (T-GNNs) are powerful representation learning methods and have achieved remarkable effectiveness on continuous-time dynamic graphs. However, T-GNNs still suffer from high time complexity, which increases linearly with the number of timestamps and grows exponentially with the model depth, causing them not scalable to large dynamic graphs. To address the limitations, we propose Orca, a novel framework that accelerates T-GNN training by non-trivially caching and reusing intermediate embeddings. We design an optimal cache replacement algorithm, named MRU, under a practical cache limit. MRU not only improves the efficiency of training T-GNNs by maximizing the number of cache hits but also reduces the approximation errors by avoiding keeping and reusing extremely stale embeddings. Meanwhile, we develop profound theoretical analyses of the approximation error introduced by our reuse schemes and offer rigorous convergence guarantees. Extensive experiments have validated that Orca can obtain two orders of magnitude speedup over the state-of-the-art baselines while achieving higher precision on large dynamic graphs. Yanyan Shen, Lei Chen 0002, Mingxuan Yuan |
Proc. ACM Manag. Data | 2 |
| 2023 | DUCATI: A Dual-Cache Training System for Graph Neural Networks on Giant Graphs with the GPUabstractRecently Graph Neural Networks (GNNs) have achieved great success in many applications. The mini-batch training has become the de-facto way to train GNNs on giant graphs. However, the mini-batch generation task is extremely expensive which slows down the whole training process. Researchers have proposed several solutions to accelerate the mini-batch generation, however, they (1) fail to exploit the locality of the adjacency matrix, (2) cannot fully utilize the GPU memory, and (3) suffer from the poor adaptability to diverse workloads. In this work, we propose DUCATI, aDual-Cache system to overcome these drawbacks. In addition to the traditionalNfeat-Cache, DUCATI introduces a newAdj-Cache to further accelerate the mini-batch generation and better utilize GPU memory. DUCATI develops a workload-awareDual-Cache Allocator which adaptively finds the best cache allocation plan under different settings. We compare DUCATI with various GNN training systems on four billion-scale graphs under diverse workload settings. The experimental results show that in terms of training time, DUCATI can achieve up to 3.33 times speedup (2.07 times on average) compared to DGL and up to 1.54 times speedup (1.32 times on average) compared to the state-of-the-artSingle-Cache systems. We also analyze the time-accuracy trade-offs of DUCATI and four state-of-the-art GNN training systems. The analysis results offer users some guidelines on system selection regarding different input sizes and hardware resources. Xin Zhang 0101, Yanyan Shen, Yingxia Shao, Lei Chen 0002 |
Proc. ACM Manag. Data | 2 |
| 2023 | Zebra: When Temporal Graph Neural Networks Meet Temporal Personalized PageRankabstractTemporal graph neural networks (T-GNNs) are state-of-the-art methods for learning representations over dynamic graphs. Despite the superior performance, T-GNNs still suffer from high computational complexity caused by the tedious recursive temporal message passing scheme, which hinders their applicability to large dynamic graphs. To address the problem, we build the theoretical connection between the temporal message passing scheme adopted by T-GNNs and the temporal random walk process on dynamic graphs. Our theoretical analysis indicates that it would be possible to select a few influential temporal neighbors to compute a target node's representation without compromising the predictive performance. Based on this finding, we propose to utilize T-PPR, a parameterized metric for estimating the influence score of nodes on evolving graphs. We further develop an efficient single-scan algorithm to answer the top- k T-PPR query with rigorous approximation guarantees. Finally, we present Zebra, a scalable framework that accelerates the computation of T-GNN by directly aggregating the features of the most prominent temporal neighbors returned by the top- k T-PPR query. Extensive experiments have validated that Zebra can be up to two orders of magnitude faster than the state-of-the-art T-GNNs while attaining better performance. Yanyan Shen, Lei Chen 0002, Mingxuan Yuan |
Proc. VLDB Endow. | 2 |
| 2023 | AutoSrh: An Embedding Dimensionality Search Framework for Tabular Data PredictionabstractPrediction over tabular data is often a crucial task in many real-life applications. Recent advances in deep learning give rise to various deep models for tabular data prediction. A common and essential step in these models is to vectorize raw input features in tabular data into dense embeddings. Choosing a suitable dimension for each feature is challenging yet necessary to improve model’s performance and reduce memory cost of model parameters. Existing solutions to embedding dimensionality search always choose dimensions from a restricted candidate set. This restriction improves the search efficiency but would produce suboptimal embedding dimensions that hurt model’s predictive performance. In this paper, we develop AutoSrh, a flexible embedding dimensionality search framework that can select varying dimensions for different features through differentiable optimization. The key idea of AutoSrh is to relax the search space to be continuous and optimize the selection of embedding dimensions via gradient descent. After optimization, AutoSrh performs embedding pruning to derive the mixed embedding dimensions and retrains the model to further improve the performance. Extensive experiments on five real-world tabular datasets demonstrate that AutoSrh can achieve better predictive performance than the existing approaches with 1.1$\sim$1.6x lower training time cost and reserve model’s predictive performance while reducing 50$\sim$95% embedding parameters. Shuming Kong, Weiyu Cheng, Yanyan Shen, Linpeng Huang |
IEEE Trans. Knowl. Data Eng. | 3 |
| 2023 | Fine Perceptive GANs for Brain MR Image Super-Resolution in Wavelet DomainabstractMagnetic resonance (MR) imaging plays an important role in clinical and brain exploration. However, limited by factors such as imaging hardware, scanning time, and cost, it is challenging to acquire high-resolution MR images clinically. In this article, fine perceptive generative adversarial networks (FP-GANs) are proposed to produce super-resolution (SR) MR images from the low-resolution counterparts. By adopting the divide-and-conquer scheme, FP-GANs are designed to deal with the low-frequency (LF) and high-frequency (HF) components of MR images separately and parallelly. Specifically, FP-GANs first decompose an MR image into LF global approximation and HF anatomical texture subbands in the wavelet domain. Then, each subband generative adversarial network (GAN) simultaneously concentrates on super-resolving the corresponding subband image. In generator, multiple residual-in-residual dense blocks are introduced for better feature extraction. In addition, the texture-enhancing module is designed to trade off the weight between global topology and detailed textures. Finally, the reconstruction of the whole image is considered by integrating inverse discrete wavelet transformation in FP-GANs. Comprehensive experiments on the MultiRes_7T and ADNI datasets demonstrate that the proposed model achieves finer structure recovery and outperforms the competing methods quantitatively and qualitatively. Moreover, FP-GANs further show the value by applying the SR results in classification tasks. Senrong You, Bai Ying Lei, Shuqiang Wang, Charles K. Chui, Albert C. Cheung, Yong Liu 0018, Min Gan, Guo-Cheng Wu 0001, Yanyan Shen |
IEEE Trans. Neural Networks Learn. Syst. | 9 |
| 2023 | Morphological Feature Visualization of Alzheimer's Disease via Multidirectional Perception GANabstractThe diagnosis of early stages of Alzheimer's disease (AD) is essential for timely treatment to slow further deterioration. Visualizing the morphological features for early stages of AD is of great clinical value. In this work, a novel multidirectional perception generative adversarial network (MP-GAN) is proposed to visualize the morphological features indicating the severity of AD for patients of different stages. Specifically, by introducing a novel multidirectional mapping mechanism into the model, the proposed MP-GAN can capture the salient global features efficiently. Thus, using the class discriminative map from the generator, the proposed model can clearly delineate the subtle lesions via MR image transformations between the source domain and the predefined target domain. Besides, by integrating the adversarial loss, classification loss, cycle consistency loss, and L1 penalty, a single generator in MP-GAN can learn the class discriminative maps for multiple classes. Extensive experimental results on Alzheimer's Disease Neuroimaging Initiative (ADNI) dataset demonstrate that MP-GAN achieves superior performance compared with the existing methods. The lesions visualized by MP-GAN are also consistent with what clinicians observe. Bai Ying Lei, Shuqiang Wang, Yong Liu 0018, Zhiguang Feng, Yong Hu 0003, Yanyan Shen, Michael Kwok-Po Ng |
IEEE Trans. Neural Networks Learn. Syst. | 7 |
| 2023 | RESUS: Warm-up Cold Users via Meta-learning Residual User Preferences in CTR PredictionabstractClick-through Rate (CTR) prediction on cold users is a challenging task in recommender systems. Recent researches have resorted to meta-learning to tackle the cold-user challenge, which either perform few-shot user representation learning or adopt optimization-based meta-learning. However, existing methods suffer from information loss or inefficient optimization process, and they fail to explicitly model global user preference knowledge, which is crucial to complement the sparse and insufficient preference information of cold users. In this article, we propose a novel and efficient approach named RESUS, which decouples the learning of global preference knowledge contributed by collective users from the learning of residual preferences for individual users. Specifically, we employ a shared predictor to infer basis user preferences, which acquires global preference knowledge from the interactions of different users. Meanwhile, we develop two efficient algorithms based on the nearest neighbor and ridge regression predictors, which infer residual user preferences via learning quickly from a few user-specific interactions. Extensive experiments on three public datasets demonstrate that our RESUS approach is efficient and effective in improving CTR prediction accuracy on cold users, compared with various state-of-the-art methods. Yanyan Shen, Lifan Zhao, Weiyu Cheng, Zibin Zhang, Kangyi Lin |
ACM Trans. Inf. Syst. | 1 |
| 2022 | Tab-HGNN: Learning Column Representation with Heterogeneous Graph Neural Network for Web Table InterpretationabstractWeb tables are valuable resources that can enrich existing knowledge bases. Researchers try to make the predictions of columns and column pairs in web tables into entities and relations of knowledge bases, which are known as the Column Type Annotation (CT) and Relation Extraction (RE) tasks in web table interpretation. Learning useful and accurate column vector representations plays the central role in solving the two tasks. A column’s semantics can be determined by three kinds of relationships: Column-Cell, Column-Table, Column-Column relationships. Existing works only rely on the Column-Cell relationship to determine the semantics of columns and result in suboptimal performances. In this paper, we propose to solve the above CT and RE tasks with the heterogeneous graph neural network technique. First, we construct heterogeneous graphs for web tables. Next, a new model named Tab-HGNN is proposed to consider all the three kinds of relationships and learn target column representations. Extensive experiments on real-world web table datasets demonstrate the effectiveness of the proposed Tab-HGNN model on the CT and RE tasks. It outperforms the competitive baselines by achieving up to +7.9% and +1.6% macro F1scores, +2.6% and +0.2% weighted F1scores on two CT datasets, respectively. It also achieves +0.3% macro F1scores, +0.1% weighted F1scores on one RE dataset. Xingyu Su, Yanyan Shen |
IEEE Big Data | 2 |
| 2022 | GCF-RD: A Graph-based Contrastive Framework for Semi-Supervised Learning on Relational DatabasesabstractRelational databases are the main storage model of structured data in most businesses, which usually involves multiple tables with key-foreign-key relationships. In practice, data analysts often want to pose predictive classification queries over relational databases. To answer such queries, many existing approaches perform supervised learning to train classification models, which heavily rely on the availability of sufficient labeled data. In this paper, we propose a novel graph-based contrastive framework for semi-supervised learning on relational databases, achieving promising predictive classification performance with only a handful of labeled data. Our framework utilizes contrastive learning to exploit additional supervision signals from massive unlabeled data. Specifically, we develop two contrastive graph views that are 1) advantageous for modeling complex relationships and correlations among structured data in a relational database, and 2) complementary to each other for learning robust representations of structured data to be classified. We also leverage label information in contrastive learning to mitigate its negative effect in knowledge transfer on the supervised counterpart. We conduct extensive experiments on three real-world relational databases and the results demonstrate that our framework is able to achieve the state-of-the-art predictive performance in limited labeled data settings, compared with various supervised and semi-supervised learning approaches. Runjin Chen, Tong Li 0017, Yanyan Shen, Luyu Qiu, Kaidi Li, Caleb Chen Cao |
CIKM | 3 |
| 2022 | Feature-Oriented Sampling for Fast and Scalable GNN TrainingabstractRecently Graph Neural Networks (GNNs) have achieved great success in many applications. To apply GNNs to large graphs, mini-batch training and sampling are widely adopted by recent works. However, existing works generate mini-batches following a topology-oriented sampling style, which first samples a subgraph and then fetches the corresponding node features to construct a mini-batch. This inevitably incurs intensive random access of graph data, the exponential growth of the batch size, and constrained candidates during sampling. In this work, we advocate adopting a feature-oriented sampling style which can overcome these drawbacks. We first sample the node features and then induce the corresponding subgraph to form a mini-batch. We apply the feature-oriented sampling method to three mainstream GNN models to demonstrate the effectiveness and efficiency of this sampling style. Experiments on four large-scale datasets show that feature-oriented sampling can achieve comparable accuracy as topology-oriented sampling while speeding up the training procedure by 2.2 ~ 7.9 times. Xin Zhang 0101, Yanyan Shen, Lei Chen 0002 |
ICDM | 2 |
| 2022 | Resolving Training Biases via Influence-based Data Relabeling
Shuming Kong, Yanyan Shen, Linpeng Huang |
ICLR | 2 |
| 2022 | Adversarial Learning Based Structural Brain-Network Generative Model for Analyzing Mild Cognitive Impairment
Heng Kong, Junren Pan, Yanyan Shen, Shuqiang Wang |
PRCV (2) | 3 |
| 2022 | Time-aware Multi-interest Capsule Network for Sequential RecommendationabstractIn recent years, sequential recommendation has been widely researched, which aims to predict the next item of interest based on user's previously interacted item sequence. Many works use RNN to model the user interest evolution over time. However, they typically compute a single vector as the user representation, which is insufficient to capture the variation of user diverse interests. Some non-RNN models employ the dynamic routing mechanism to automatically vote out multiple capsules that represent user's diverse interests, but they are ignorant of the temporal information of user's historical behaviors, thus yielding suboptimal performance. In this paper, we aim to establish a time-aware dynamic routing algorithm to effectively extract temporal user multiple interests for sequential recommendation. We observe that the significance of an item to user interests may change monotonically over time, and user interests may fluctuate periodically. Following the intuitive temporal patterns of user interests, we propose a novel time-aware multi-interest capsule network named TAMIC that leverages two kinds of time-aware voting gates, i.e., monotonic gates and periodic gates, to control the influence of each interacted item on user's current interests during the routing procedure. We further employ an aggregation module to form a temporal multi-interest user representation which is used for next item prediction. Extensive experiments on real-world datasets verify the effectiveness of the time gates and the superior performance of our TAMIC approach on sequential recommendation, compared with the state-of-the-art methods. Yanyan Shen |
SDM | 2 |
| 2022 | Camel: Managing Data for Efficient Stream LearningabstractMany real-world applications rely on predictive models that are incrementally learned online. Specifically, models are updated with a single pass over continuously arriving data batches in a typical stream learning framework. However, this framework has three shortcomings: high training cost, low data effectiveness, and catastrophic forgetting. We describe Camel, a system that addresses the above issues. Camel includes two independent data management components: coreset selection and buffer update. To accelerate model training, Camel selects a coreset from each streaming data batch for model update. Selecting a coreset with worst-case guarantees is NP-hard. To solve this problem, we reformulate coreset selection as a submodular maximization problem by deriving an upper bound on the objective function. To mitigate catastrophic forgetting, Camel maintains a buffer of past representative samples as new data arrive. Moreover, Camel quantizes numerical data in buffer via a quantile sketch to reduce the memory footprint. Finally, extensive experiments validate the effectiveness and efficiency of Camel. In particular, our coreset selection algorithm can achieve a linear speedup with a marginal accuracy loss on redundant datasets. Furthermore, our buffer update algorithms can outperform the state-of-the-art methods for anti-forgetting on various data distributions. Yanyan Shen, Lei Chen 0002 |
SIGMOD Conference | 2 |
| 2022 | HAP: An Efficient Hamming Space Index Based on Augmented Pigeonhole PrincipleabstractThe emerging deep learning techniques prefer mapping complex data objects (e.g., images, documents) to compact binary vectors (i.e., hash codes) for efficient similarity search. In this paper, we study the problem of indexing large-scale binary databases to support fast Hamming distance-based similarity queries. Existing Hamming space indices usually divide long binary vectors into short disjoint pieces and apply the Pigeonhole Principle to prune unnecessary candidates. In our work, we relax the disjoint partition constraint by allowing dimension redundancy, which yields a tighter pruning bound named Augmented Pigeonhole Principle (APP). Intuitively, APP enables more optimization opportunities by capturing the correlation between database and query workloads. Based on APP, we propose HAP, an efficient Hamming space index framework to support both Hamming range queries and k-NN queries. Qiyu Liu, Yanyan Shen, Lei Chen 0002 |
SIGMOD Conference | 2 |
| 2022 | Cybertwin-Driven Multi-Intelligent Reflecting Surfaces aided Vehicular Edge Computing Leveraged by Deep Reinforcement LearningabstractRecently, the cybertwin-driven intelligent internet of vehicles has received widespread consideration in modern smart cities which makes it possible to run high dimensional, low-latency tolerating, and computational-intensive tasks on the vehicles. Thanks to the development in mobile edge computing, the so-called vehicular edge computing allows mobile vehicles to offload their tasks to the road-side unit or hybrid access point due to the limited computation capability. In this paper, we consider a cybertwin-driven internet of vehicle system that provides computing services for mobile vehicles in local area network or wide area network aided with multi-intelligent reflecting surfaces. Based on this system model, we investigate an optimization problem to jointly maximize the sum of data rate in wide area network, and the sum of energy utilities of vehicles. However, in the proposed system model, it is complicated to design the optimal phase, scheduling and offloading decision policy. To solve this issue, we propose a block coordinate descent and deep reinforcement learning based intelligent IoV computing policy. Numerical results have verified that the proposed algorithm can achieve better IoV computing performance compared with four relative benchmark algorithms. Huijun Xing, Weilin Zang, Zhenzhen Jin, Yanyan Shen |
VTC Fall | 5 |
| 2022 | LECF: recommendation via learnable edge collaborative filtering
Shitao Xiao, Yingxia Shao, Yawen Li 0001, Hongzhi Yin, Yanyan Shen, Bin Cui 0001 |
Sci. China Inf. Sci. | 5 |
| 2022 | Brain stroke lesion segmentation using consistent perception generative adversarial network
Shuqiang Wang, Senrong You, Bingchuan Wang, Yanyan Shen, Bai Ying Lei |
Neural Comput. Appl. | 5 |
| 2022 | SANCUS: Staleness-Aware Communication-Avoiding Full-Graph Decentralized Training in Large-Scale Graph Neural NetworksabstractGraph neural networks (GNNs) have emerged due to their success at modeling graph data. Yet, it is challenging for GNNs to efficiently scale to large graphs. Thus, distributed GNNs come into play. To avoid communication caused by expensive data movement between workers, we propose Sancus, a staleness-aware communication-avoiding decentralized GNN system. By introducing a set of novel bounded embedding staleness metrics and adaptively skipping broadcasts, Sancus abstracts decentralized GNN processing as sequential matrix multiplication and uses historical embeddings via cache. Theoretically, we show bounded approximation errors of embeddings and gradients with convergence guarantee. Empirically, we evaluate Sancus with common GNN models via different system setups on large-scale benchmark datasets. Compared to SOTA works, Sancus can avoid up to 74% communication with at least 1.86X faster throughput on average without accuracy loss. Jingshu Peng, Zhao Chen 0003, Yingxia Shao, Yanyan Shen, Lei Chen 0002, Jiannong Cao 0001 |
Proc. VLDB Endow. | 4 |
| 2022 | MBN: Towards Multi-Behavior Sequence Modeling for Next Basket RecommendationabstractNext basket recommendation aims at predicting the next set of items that a user would likely purchase together, which plays an important role in e-commerce platforms. Unlike conventional item recommendation, the next basket recommendation focuses on capturing item correlations among baskets and learning the user’s temporal interest from the past purchasing basket sequence. In practice, most users interact with items in various kinds of behaviors. The multi-behavior data sheds light on user’s potential purchasing intention and resolves noisy signals from accidentally purchased items. In this article, we conduct an empirical study on real datasets to exploit the characteristics of multi-behavior data and confirm its positive effects on next basket recommendation. We develop a novel Multi-Behavior Network (MBN) model that captures item correlations and acquires meta-knowledge from multi-behavior basket sequences effectively. MBN employs the meta multi-behavior sequence encoder to model temporal dependencies of each individual behavior and extract meta-knowledge across different behaviors. Furthermore, we design the recurring-item-aware predictor in MBN to realize the high degree of the repeated occurrences of items, leading to better recommendation performance. We conduct extensive experiments to evaluate the performance of our proposed MBN model using real-world multi-behavior data. The results demonstrate the superior recommendation performance of MBN compared with various state-of-the-art methods. Yanyan Shen, Baoyuan Ou, Ranzhen Li |
ACM Trans. Knowl. Discov. Data | 1 |
| 2022 | Bidirectional Mapping Generative Adversarial Networks for Brain MR to PET SynthesisabstractFusing multi-modality medical images, such as magnetic resonance (MR) imaging and positron emission tomography (PET), can provide various anatomical and functional information about the human body. However, PET data is not always available for several reasons, such as high cost, radiation hazard, and other limitations. This paper proposes a 3D end-to-end synthesis network called Bidirectional Mapping Generative Adversarial Networks (BMGAN). Image contexts and latent vectors are effectively used for brain MR-to-PET synthesis. Specifically, a bidirectional mapping mechanism is designed to embed the semantic information of PET images into the high-dimensional latent space. Moreover, the 3D Dense-UNet generator architecture and the hybrid loss functions are further constructed to improve the visual quality of cross-modality synthetic images. The most appealing part is that the proposed method can synthesize perceptually realistic PET images while preserving the diverse brain structures of different subjects. Experimental results demonstrate that the performance of the proposed method outperforms other competitive methods in terms of quantitative measures, qualitative displays, and evaluation metrics for classification. Shengye Hu, Bai Ying Lei, Shuqiang Wang, Yong Wang 0002, Zhiguang Feng, Yanyan Shen |
IEEE Trans. Medical Imaging | 6 |
| 2022 | Tensorizing GAN With High-Order Pooling for Alzheimer's Disease AssessmentabstractIt is of great significance to apply deep learning for the early diagnosis of Alzheimer's disease (AD). In this work, a novel tensorizing GAN with high-order pooling is proposed to assess mild cognitive impairment (MCI) and AD. By tensorizing a three-player cooperative game-based framework, the proposed model can benefit from the structural information of the brain. By incorporating the high-order pooling scheme into the classifier, the proposed model can make full use of the second-order statistics of holistic magnetic resonance imaging (MRI). To the best of our knowledge, the proposed Tensor-train, High-order pooling and Semisupervised learning-based GAN (THS-GAN) is the first work to deal with classification on MR images for AD diagnosis. Extensive experimental results on Alzheimer's disease neuroimaging initiative (ADNI) data set are reported to demonstrate that the proposed THS-GAN achieves superior performance compared with existing methods, and to show that both tensor-train and high-order pooling can enhance classification performance. The visualization of generated samples also shows that the proposed model can generate plausible samples for semisupervised learning purpose. Bai Ying Lei, Michael Kwok-Po Ng, Albert C. Cheung, Yanyan Shen, Shuqiang Wang |
IEEE Trans. Neural Networks Learn. Syst. | 5 |
| 2022 | An Ensemble-Based Densely-Connected Deep Learning System for Assessment of Skeletal MaturityabstractAssessment of skeletal maturity is important for a clinician to make a decision of the most appropriate treatment on various skeletal disorders. This task is very challenging when using machine learning method due to the limited data and large anatomical variations among different subjects. In this article, we propose an ensemble-based deep learning pipeline to automatically assess the distal radius and ulna (DRU) maturity from left-hand radiographs. At the same time, we adapted the concept of densely connected mechanism in the proposed network architecture to reuse features and prevent gradient disappearance. Therefore, the model acquires two convincing advantages: first, our model preserves the maximum information flow and has a much faster convergence rate. Second, our model avoids overfitting even if training with limited data. The experimental dataset contains 1189 left-hand$X$-ray scans of children and teenagers. The proposed method achieves 85.27% and 91.68% for radius and ulna classification, respectively. Extensive experiments prove that our model performs better than using other network structures. Shuqiang Wang, Yanyan Shen, Prudence Wing-Hang Cheung, Jason Pui Yin Cheung, Keith D. K. Luk, Yong Hu 0003 |
IEEE Trans. Syst. Man Cybern. Syst. | 3 |
| 2022 | Special issue on responsible data management and data science
Zi Huang, Yanyan Shen, Divesh Srivastava |
VLDB J. | 2 |
| 2021 | Palette: Towards Multi-source Model Selection and Ensemble for ReuseabstractThe success of deep learning presents exciting opportunities for reusing pre-trained models from source domains to solve a target task with possibly limited training data. While various pre-trained models have been developed and become available, there is no principled way to select appropriate models for reuse. Although it is suggested that source and target tasks should be sufficiently similar, the calculation of task relevance usually requires extra storage of source training data and great efforts from domain experts, which is impractical in many applications. In this paper, we study the Multi-source Model Selection and Ensemble (MSMSE) problem. Given a collection of source models, we aim to select a subset of source models and develop an ensemble model that achieves the best performance for a target task. An ensemble of multiple models enables a boarder utilization of various underlying source knowledge and leads to better generalization ability. To this end, we present Palette, a generic framework that first selects potentially well-performed models from a source model pool, and then builds an ensemble with refinement. We introduce different model selection strategies that combine multi-armed bandits with adaptive resource allocation and Bayesian optimization techniques to accelerate the selection process. Extensive experimental results validate the effectiveness and efficiency of Palette and our model selection strategies. Yanyan Shen, Lei Chen 0002 |
ICDE | 2 |
| 2021 | LHist: Towards Learning Multi-dimensional Histogram for Massive Spatial DataabstractData synopsis is widely adopted to speed-up query processing over large spatial databases. As one of the most popular spatial data synopses, multi-dimensional histograms (MH) have been studied and adopted by modern DBMS and analytical systems for decades. However, existing MH construction techniques highly rely on expert knowledge and statistical assumptions, making them hard to achieve consistently satisfactory performance across different datasets. Inspired by the emerging learned index techniques where the widely used index structures like B-tree can be further improved by integrating simple machine learning models, in this paper, we propose a learned data synopsis technique named Learned Multi-dimensional Histogram (LHist). Compared with the traditional data synopsis techniques, LHist is fully data-driven, easy-to-implement, and has the potential to achieve better storage-accuracy trade-off. On the typical task of range COUNT query estimation, the extensive experimental studies on large-scale real-world datasets and synthetic benchmarks reveal that LHist can outperform the existing synopsis structures in terms of storage cost, query processing efficiency, and estimation accuracy. Qiyu Liu, Yanyan Shen, Lei Chen 0002 |
ICDE | 2 |
| 2021 | Mcore: Multi-Agent Collaborative Learning for Knowledge-Graph-Enhanced RecommendationabstractRecently, knowledge-graph-enhanced recommendation systems have attracted much attention, since knowledge graph (KG) can help improving the dataset quality and offering rich semantics for explainable recommendation. However, current KG-enhanced solutions focus on analyzing user behaviors on the product level and lack effective approaches to extract user preference towards product category, which is essential for better recommendation because users shopping online normally have strong preference towards distinctive product categories, not merely on products, according to various user studies. Moreover, the existing pure embedding-based recommendation methods can only utilize KGs with a limited size, which is not adaptable to many real-world applications. In this paper, we generalize the recommendation problem with preference mining as a compound knowledge reasoning task and propose a novel multi-agent system, called Mcore, which can promote model performance by mining users’ high-level interests and is adaptable to large KGs. Specifically, we split the overall problem and allocate sub-task to each agent: Coordinate Agent takes charge of recognizing the product-category preference of current user, while Relation Agent and Entity Agent perform KG reasoning cooperatively from a user node towards the preferred categories and terminate at a product node as recommendation. To train this heterogeneous multi-agent system, where agents own various functionalities, we propose an asynchronous reinforcement training pipeline, called Multi-agent Collaborative Learning. The extensive experiments on real datasets demonstrate the effectiveness and adaptability of Mcore on recommendation tasks. Xujia Li, Yanyan Shen, Lei Chen 0002 |
ICDM | 2 |
| 2021 | GraphANGEL: Adaptive aNd Structure-Aware Sampling on Graph NEuraL NetworksabstractGraph neural networks (GNNs) have gained increasing attention in their great success at graph representation learning. In GNNs, the neighborhood of the target node is aggregated iteratively to capture and learn its local structure and neighbor information. Observing that different nodes often require a distinct number of iterations to better learn the representation, we propose an adaptive and structure-aware graph sampling scheme GraphANGEL for GNNs. However, it is quite challenging because both the suitable range of exploration and the important substructure in the neighborhood are difficult to determine. Exploiting the unique feature of random walk mixing time and various node structural role importance measures, we first propose a lightweight component to flexibly estimate the proper neighborhood exploration depth for each target node. Then we investigate different importance metrics to identify and sample the most structurally critical subgraphs that carry a larger influence in messaging passing. Moreover, since different importance metrics unveil different aspects of the graph, we combine and ensemble various importance measures with attention to boost the final performance. In this manner, our method adaptively and explicitly embeds the structural importance information of a node and its critical neighborhood at the same time for finer structure-aware graph representation learning. Evaluation on the benchmark datasets suggests the competitive performance of GraphANGEL to the state-of-the-art approaches, demonstrating the effectiveness of our adaptive and structure-aware sampling approach. Jingshu Peng, Yanyan Shen, Lei Chen 0002 |
ICDM | 2 |
| 2021 | A Point Cloud Generative Model via Tree-Structured Graph Convolutions for 3D Brain Shape Reconstruction
Bai Ying Lei, Yanyan Shen, Yong Liu 0018, Shuqiang Wang |
PRCV (2) | 3 |
| 2021 | Characterization Multimodal Connectivity of Brain Network by Hypergraph GAN for Alzheimer's Disease Analysis
Junren Pan, Bai Ying Lei, Yanyan Shen, Yong Liu 0018, Zhiguang Feng, Shuqiang Wang |
PRCV (3) | 3 |
| 2021 | Multimodal Representations Learning and Adversarial Hypergraph Fusion for Early Alzheimer's Disease Prediction
Qiankun Zuo, Bai Ying Lei, Yanyan Shen, Yong Liu 0018, Zhiguang Feng, Shuqiang Wang |
PRCV (3) | 3 |
| 2021 | DRL based Data Offloading for Intelligent Reflecting Surface Aided Mobile Edge ComputingabstractRecently, the intelligent reflecting surface (IRS) is an emerging and promising technology for achieving higher spectrum and energy efficiency in wireless communication systems. In this paper, we consider a wireless powered mobile edge computing (MEC) network that is equipped with an IRS. The IRS is able to provide a reflecting channel to enhance the offloading capability for edge users. Based on this system model, we investigate an optimisation problem to maximize the sum of users' utilities, which jointly consider the energy efficiency, time latency, and price of offloading computations. With task offloading, power limited users can complete the computational tasks even when they face data-intensive workloads. However, in a dynamic system, it is complicated to design the optimal offloading decision strategy. To tackle this problem, we propose a deep reinforcement learning (DRL) based approach. In the designed algorithm, in order to get a better reward, the agent chooses a near optimal solution to adjust the workload partitions, the time allocation, and IRS parameters according to the dynamic channel environment and the random arrival of task workload. Numerical results show that the proposed DRL based IRS-aided offloading algorithm can achieve better system performance compared with that without IRS and the relative benchmark algorithms. Yanyan Shen, Bo Yang 0006, Weilin Zang, Shuqiang Wang |
WCNC | 2 |
| 2021 | FluxEV: A Fast and Effective Unsupervised Framework for Time-Series Anomaly DetectionabstractAnomaly detection in time series is a research area of increasing importance. In order to safeguard the availability and stability of services, large companies need to monitor various time-series data to detect anomalies in real time for troubleshooting, thereby reducing potential economic losses. However, in many practical applications, time-series anomaly detection is still an intractable problem due to the huge amount of data, complex data patterns, and limited computational resources. SPOT is an efficient streaming algorithm for anomaly detection, but it is only sensitive to extreme values in the whole data distribution. In this paper, we propose FluxEV, a fast and effective unsupervised anomaly detection framework. By converting the non-extreme anomalies to extreme values, our framework addresses the limitation of SPOT and achieves a huge improvement in the detection accuracy. Moreover, Method of Moments is adopted to speed up the parameter estimation in the automatic thresholding. Extensive experiments show that FluxEV greatly outperforms the state-of-the-art baselines on two large public datasets while ensuring high efficiency. Jia Li 0014, Shimin Di, Yanyan Shen, Lei Chen 0002 |
WSDM | 3 |
| 2021 | GNEM: A Generic One-to-Set Neural Entity Matching FrameworkabstractEntity Matching is a classic research problem in any data analytics pipeline, aiming to identify records referring to the same real-world entity. It plays an important role in data cleansing and integration. Advanced entity matching techniques focus on extracting syntactic or semantic features from record pairs via complex neural architectures or pre-trained language models. However, the performances always suffer from noisy or missing attribute values in the records. We observe that comparing one record with several relevant records in a collective manner allows each pairwise matching decision to be made by borrowing valuable insights from other pairs, which is beneficial to the overall matching performance. In this paper, we propose a generic one-to-set neural framework named GNEM for entity matching. GNEM predicts matching labels between one record and a set of relevant records simultaneously. It constructs a record pair graph with weighted edges and adopts the graph neural network to propagate information among pairs. We further show that GNEM can be interpreted as an extension and generalization of the existing pairwise matching techniques. Extensive experiments on real-world data sets demonstrate that GNEM consistently outperforms the existing pairwise entity matching techniques and achieves up to 8.4% improvement on F1-Score compared with the state-of-the-art neural methods. Runjin Chen, Yanyan Shen, Dongxiang Zhang |
WWW | 2 |
| 2021 | Robust energy efficient maximization in wireless powered CRNs based on power splitting
Zhixin Liu 0001, Meihua Zhou, Yanyan Shen, Yazhou Yuan, Kit Yan Chan, Yi Yang 0030 |
Comput. Networks | 3 |
| 2021 | Joint 3-D Trajectory and Resource Optimization in Multi-UAV-Enabled IoT Networks With Wireless Power TransferabstractThis article studies the data collection problem in an Internet-of-Things (IoT) network with multiple unmanned aerial vehicles (UAVs) where UAVs first power multiple IoT devices by wireless power transfer, and then IoT devices utilize the harvested energy to transmit data to UAVs. Different from most of the existing works that often assume the channel between the UAV and the IoT device is a simplified Line-of-Sight (LoS) channel, a more practical and accurate probabilistic LoS channel model is adopted, in which both the elevation angle and the distance between the UAV and the IoT device determine the channel gain. Our objective is to maximize the UAV's minimum data collection rate among all IoT devices by jointly optimizing time allocation and 3-D trajectory of UAVs within a limited time duration. This results in a nonconvex optimization problem, which is challenge to solve. To tackle this difficulty, we transform the nonconvex problem to a difference of convex (D.C.) optimization problem by subtly using several methods. To solve the D.C. optimization problem, an efficient iterative algorithm is designed via a successive convex approximation method. Numerical simulation results are provided to verify the performance of the proposed algorithm compared to two benchmark algorithms, the algorithm with simplified LoS model and that with 2-D trajectory optimization, under various conditions. Weiran Luo, Yanyan Shen, Bo Yang 0006, Shuqiang Wang, Xin-Ping Guan |
IEEE Internet Things J. | 2 |
| 2021 | ETO: Accelerating Optimization of DNN Operators by High-Performance Tensor Program ReuseabstractRecently, deep neural networks (DNNs) have achieved great success in various applications, where low inference latency is important. Existing solutions either manually tune the kernel library or utilize search-based compilation to reduce the operator latency. However, manual tuning requires significant engineering effort, and the huge search space makes the search cost of the search-based compilation unaffordable in some situations. In this work, we propose ETO, a framework for speeding up DNN operator optimization based on reusing the information of performant tensor programs. Specifically, ETO defines conditions for the information reuse between two operators. For operators satisfying the conditions, based on the performant tensor program information of one operator, ETO uses a reuse-based tuner to significantly prune the search space of the other one, and keeps optimization effectiveness at the same time. In this way, for a set of operators, ETO first determines the information reuse relationships among them to reduce the total search time needed, and then tunes the operators either by the backend compiler or by the reuse-based tuner accordingly. ETO further increases the reuse opportunities among the operators by injecting extra operators as bridges between two operators which do not satisfy the reuse conditions. Compared with various existing methods, the experiments show that ETO is effective and efficient in optimizing DNN operators. Jingzhi Fang, Yanyan Shen, Yue Wang 0012, Lei Chen 0002 |
Proc. VLDB Endow. | 2 |
| 2021 | Diabetic Retinopathy Diagnosis Using Multichannel Generative Adversarial Network With SemisupervisionabstractDiabetic retinopathy (DR) is one of the major causes of blindness. It is of great significance to apply deep-learning techniques for DR recognition. However, deep-learning algorithms often depend on large amounts of labeled data, which is expensive and time-consuming to obtain in the medical imaging area. In addition, the DR features are inconspicuous and spread out over high-resolution fundus images. Therefore, it is a big challenge to learn the distribution of such DR features. This article proposes a multichannel-based generative adversarial network (MGAN) with semisupervision to grade DR. The multichannel generative model is developed to generate a series of subfundus images corresponding to the scattering DR features. By minimizing the dependence on labeled data, the proposed semisupervised MGAN can identify the inconspicuous lesion features by using high-resolution fundus images without compression. Experimental results on the public Messidor data set show that the proposed model can grade DR effectively. Note to Practitioners-This article is motivated by the challenging problem due to the inadequacy of labeled data in medical image analysis and the dispersion of efficient features in high-resolution medical images. As for the inadequacy of labeled data in medical image analysis, the reasons mainly include the followings: 1) the high-quality annotation of medical imaging sample depends heavily on scarce medical expertise which is very expensive and 2) comparing with natural issues, it is more difficult to collect medical images because of privacy issues. It is of great significance to apply deep-learning techniques for diabetic retinopathy (DR) recognition. In this article, the multichannel generative adversarial network (GAN) with semisupervision is developed for DR-aided diagnosis. The proposed model can deal with DR classification problem with inadequacy of labeled data in the following ways: 1) the multichannel generative scheme is proposed to generate a series of subfundus images corresponding to the scattering DR features and 2) the proposed multichannel-based GAN (MGAN) model with semisupervision can make full use of both labeled data and unlabeled data. The experimental results demonstrate that the proposed model outperforms the other representative models in terms of accuracy, area under ROC curve (AUC), sensitivity, and specificity. Shuqiang Wang, Yong Hu 0003, Yanyan Shen, Zhile Yang, Min Gan, Bai Ying Lei |
IEEE Trans Autom. Sci. Eng. | 4 |
| 2021 | Dual-Embedding based Deep Latent Factor Models for RecommendationabstractAmong various recommendation methods, latent factor models are usually considered to be state-of-the-art techniques, which aim to learn user and item embeddings for predicting user-item preferences. When applying latent factor models to the recommendation with implicit feedback, the quality of embeddings always suffers from inadequate positive feedback and noisy negative feedback. Inspired by the idea of NSVD that represents users based on their interacted items, this article proposes a dual-embedding based deep latent factor method for recommendation with implicit feedback. In addition to learning a primitive embedding for a user (resp. item), we represent each user (resp. item) with an additional embedding from the perspective of the interacted items (resp. users) and propose attentive neural methods to discriminate the importance of interacted users/items for dual-embedding learning. We design two dual-embedding based deep latent factor models, DELF and DESEQ, for pure collaborative filtering and temporal collaborative filtering (i.e., sequential recommendation), respectively. The novel attempt of the proposed models is to capture each user-item interaction with four deep representations that are subtly fused for preference prediction. We conducted extensive experiments on four real-world datasets. The results verify the effectiveness of user/item dual embeddings and the superior performance of our methods on item recommendation. Weiyu Cheng, Yanyan Shen, Linpeng Huang, Yanmin Zhu 0006 |
ACM Trans. Knowl. Discov. Data | 2 |
| 2021 | Multi-Level Attention Networks for Multi-Step Citywide Passenger Demands PredictionabstractFor the emerging mobility-on-demand services, it is of great significance to predict passenger demands based on historical mobility trips towards better vehicle distribution. Prior works have focused on predicting next-step passenger demands at selected locations or hotspots. However, we argue that multi-step citywide passenger demands encapsulate both time-varying demand trends and global statuses, and hence are more beneficial to avoiding demand-service mismatching and developing effective vehicle distribution/scheduling strategies. Furthermore, we find that adaptations of single-step methods are unable to achieve robust prediction with high accuracy for further steps. In this paper, we propose an end-to-end deep neural network model to the prediction task. We employ an encoder-decoder framework based on convolutional and ConvLSTM units to identify complex features that capture spatiotemporal influence and pickup-dropoff interactions on citywide passenger demands. We introduce a multi-level attention model (global attention and temporal attention) to emphasize the effects of latent citywide mobility regularities and capture relevant temporal dependencies. We evaluate our proposed method using real-world mobility trips (taxis and bikes) and the experimental results show that our method achieves higher prediction accuracy than the state-of-the-art approaches. Xian Zhou 0003, Yanyan Shen, Linpeng Huang, Tianzi Zang, Yanmin Zhu 0006 |
IEEE Trans. Knowl. Data Eng. | 2 |
| 2020 | Adaptive Factorization Network: Learning Adaptive-Order Feature InteractionsabstractVarious factorization-based methods have been proposed to leverage second-order, or higher-order cross features for boosting the performance of predictive models. They generally enumerate all the cross features under a predefined maximum order, and then identify useful feature interactions through model training, which suffer from two drawbacks. First, they have to make a trade-off between the expressiveness of higher-order cross features and the computational cost, resulting in suboptimal predictions. Second, enumerating all the cross features, including irrelevant ones, may introduce noisy feature combinations that degrade model performance. In this work, we propose the Adaptive Factorization Network (AFN), a new model that learns arbitrary-order cross features adaptively from data. The core of AFN is a logarithmic transformation layer that converts the power of each feature in a feature combination into the coefficient to be learned. The experimental results on four real datasets demonstrate the superior predictive performance of AFN against the state-of-the-arts. Weiyu Cheng, Yanyan Shen, Linpeng Huang |
AAAI | 2 |
| 2020 | DDSL: Efficient Subgraph Listing on Distributed and Dynamic Graphs
Xun Jian 0001, Yue Wang 0012, Xiayu Lei, Yanyan Shen, Lei Chen 0002 |
DASFAA (2) | 4 |
| 2020 | Finish Them on the Fly: An Incentive Mechanism for Real-Time Spatial Crowdsourcing
Qiyu Liu, Libin Zheng 0001, Yanyan Shen, Lei Chen 0002 |
DASFAA (2) | 3 |
| 2020 | FreqST: Exploiting Frequency Information in Spatiotemporal Modeling for Traffic PredictionabstractThe goal of traffic prediction is to predict future traffic values such as speeds or flows in a transportation network. Existing studies mainly focus on exploring spatiotemporal dependencies in time and space domains. However, they generally neglect the frequency information encoded in the traffic data, which sheds light on the true spatiotemporal correlations among any vertices in the transportation network. In this paper, we aim to incorporate two kinds of frequency information, time-frequency and space-frequency, into the modeling of spatiotemporal dependencies towards more accurate traffic prediction. We investigate the importance of two frequencies using real traffic data and develop an end-to-end neural approach to utilizing both frequencies for traffic prediction effectively. Specifically, we propose an attention module to aggregate features from semantically related vertices according to their time-frequency similarities, and derive a semantic representation. A novel space-frequency-based gating mechanism is designed to control how much local information is propagated to the state updating, enforced with the graph convolution to generate an effective local representation. Finally, our model combines the semantic and local representations with gated recurrent units to predict future traffic values. Extensive experiments on two real-world traffic datasets demonstrate that our proposed approach yields better prediction performance than the state-of-the-art methods. Xian Zhou 0003, Yanyan Shen, Linpeng Huang |
ICDM | 2 |
| 2020 | Intent Preference Decoupling for User Representation on Online Recommender SystemabstractAccurately characterizing the user's current interest is the core of recommender systems. However, users' interests are dynamic and affected by intent factors and preference factors. The intent factors imply users' current needs and change among different visits. The preference factors are relatively stable and learned continuously over time. Existing works either resort to the sequential recommendation to model the current browsing intent and historical preference separately or just mix up these two factors during online learning. In this paper, we propose a novel learning strategy named FLIP to decouple the learning of intent and preference under the online settings. The learning of the intent is considered as a meta-learning task and fast adaptive to the current browsing; the learning of the preference is based on the calibrated user intent and constantly updated over time. We conducted experiments on two public datasets and a real-world recommender system. When equipping it with modern recommendation methods, significant improvements are demonstrated over strong baselines. Zhaoyang Liu 0003, Fei Sun 0001, Jinyang Gao, Bolin Ding, Yanyan Shen |
IJCAI | 7 |
| 2020 | Brain MR to PET Synthesis via Bidirectional Generative Adversarial Network
Shengye Hu, Yanyan Shen, Shuqiang Wang, Bai Ying Lei |
MICCAI (2) | 2 |
| 2020 | Multi-Context Attention for Entity MatchingabstractEntity matching (EM) is a classic research problem that identifies data instances referring to the same real-world entity. Recent technical trend in this area is to take advantage of deep learning (DL) to automatically extract discriminative features. DeepER and DeepMatcher have emerged as two pioneering DL models for EM. However, these two state-of-the-art solutions simply incorporate vanilla RNNs and straightforward attention mechanisms. In this paper, we fully exploit the semantic context of embedding vectors for the pair of entity text descriptions. In particular, we propose an integrated multi-context attention framework that takes into account self-attention, pair-attention and global-attention from three types of context. The idea is further extended to incorporate attribute attention in order to support structured datasets. We conduct extensive experiments with 7 benchmark datasets that are publicly accessible. The experimental results clearly establish our superiority over DeepER and DeepMatcher in all the datasets. Dongxiang Zhang, Yuyang Nie, Sai Wu, Yanyan Shen, Kian-Lee Tan |
WWW | 4 |
| 2020 | Energy-efficient resource allocation in wireless powered CCRNs with simultaneous wireless information and power transfer
Zhixin Liu 0001, Meihua Zhou, Yanyan Shen, Kit Yan Chan, Xin-Ping Guan |
Comput. Commun. | 3 |
| 2020 | Low complexity resource allocation algorithms for chunk based OFDMA multi-user networks with max-min fairness
Yanyan Shen, Xiaoxia Huang 0004, Bo Yang 0006, Shuqiang Wang |
Comput. Commun. | 1 |
| 2020 | Resource configuration for minimizing source energy consumption in multi-carrier networks with energy harvesting relay and data-rate guarantee
Yanyan Shen, Xiaoxia Huang 0004, Kyung Sup Kwak |
Comput. Commun. | 3 |
| 2020 | DASFAA 20202 Special Issue Editorial
Yingxia Shao, Yanyan Shen, Bin Cui 0001, Jeffrey Xu Yu |
Data Sci. Eng. | 2 |
| 2020 | Hybrid multi-mode machine learning-based fault diagnosis strategies with application to aircraft gas turbine engines
Yanyan Shen, Khashayar Khorasani |
Neural Networks | 1 |
| 2020 | Optimizing DNN Computation Graph using Graph Substitutions
Jingzhi Fang, Yanyan Shen, Yue Wang 0012, Lei Chen 0002 |
Proc. VLDB Endow. | 2 |
| 2020 | Stable Learned Bloom Filters for Data Streams
Qiyu Liu, Libin Zheng 0001, Yanyan Shen, Lei Chen 0002 |
Proc. VLDB Endow. | 3 |
| 2020 | AR2Net: An Attentive Neural Approach for Business Location Selection with Satellite Data and Urban DataabstractBusiness location selection is crucial to the success of businesses. Traditional approaches like manual survey investigate multiple factors, such as foot traffic, neighborhood structure, and available workforce, which are typically hard to measure. In this article, we propose to explore both satellite data (e.g., satellite images and nighttime light data) and urban data for business location selection tasks of various businesses. We extract discriminative features from the two kinds of data and perform empirical analysis to evaluate the correlation between extracted features and the business popularity of locations. A novel neural network approach named R 2 Net is proposed to learn deep interactions among features and predict the business popularity of locations. The proposed approach is trained with a regression-and-ranking combined loss function to preserve accurate popularity estimation and the ranking order of locations simultaneously. To support the location selection for multiple businesses, we propose an approach named AR 2 Net with three attention modules, which enable the approach to focus on different latent features according to business types. Comprehensive experiments on a real-world dataset demonstrate that the satellite features are effective and our models outperform the state-of-the-art methods in terms of four metrics. Yanyan Shen, Yanmin Zhu 0006, Jiadi Yu |
ACM Trans. Knowl. Discov. Data | 2 |
| 2020 | GAT: A Unified GPU-Accelerated Framework for Processing Batch Trajectory QueriesabstractThe increasing amount of trajectory data facilitates a wide spectrum of practical applications in which large numbers of trajectory range and similarity queries are issued continuously. This calls for high-throughput trajectory query processing. Traditional in-memory databases lack considerations of the unique features of trajectories, while specialized trajectory query processing systems are typically designed for only one type of trajectory queries. This paper introduces GAT, a unified GPU-accelerated framework to process batch trajectory queries with the objective of high throughput. GAT follows the filtering-and-verification paradigm where we develop a novel index GTIDX for effectively filtering invalid trajectories on the CPU, and exploit the massive parallelism of the GPU for verification. To optimize the performance of GAT, we first greedily partition batch queries to reduce the amortized query processing latency. We then apply the Morton-based encoding method to coalesce data access requests from the GPU cores, and maintain a hash table to avoid redundant data transfer between CPU and GPU. To achieve load balance, we group size-varying cells into balanced blocks with similar numbers of trajectory points. Extensive experiments have been conducted over real-life trajectory datasets. The results show that GAT is efficient, scalable, and achieves high throughput with acceptable indexing cost. Kaixing Dong, Bowen Zhang 0006, Yanyan Shen, Yanmin Zhu 0006, Jiadi Yu |
IEEE Trans. Knowl. Data Eng. | 3 |
| 2019 | Diagnosis, Prognosis and Health Monitoring of Electro Hydraulic Servo Valves (EHSV) using Particle FiltersabstractElectro hydraulic servo valves (EHSV) constitute as core parts of many hydraulic actuators such as multi-functional spoilers (MFS) in aircraft systems. Their continuous health monitoring is important for the overall flight safety as well as economizing down times and repairs. Hence, diagnosis, prognosis and health monitoring (DPHM) of EHSVs is addressed in this paper for multiple modes of degradations that might be present in the system. The objective of this paper is to demonstrate how each mode of degradation may be isolated and estimated, and how the remaining useful life (RUL) of the EHSV can be assessed for a critically safe component under strict safety regulations. In this paper Bayesian tracking and particle filters are utilized to address three main diagnosis and prognosis problems, namely `isolation', `estimation', and the `remaining useful life'. Several case studies simulations have been provided to demonstrate and illustrate the capabilities of our proposed methodologies. Shahram Shahkar, Yanyan Shen, Khashayar Khorasani |
CoDIT | 2 |
| 2019 | A Frequency-Aware Spatio-Temporal Network for Traffic Flow Prediction
Shunfeng Peng, Yanyan Shen, Yanmin Zhu 0006, Yuting Chen 0001 |
DASFAA (2) | 2 |
| 2019 | Fault Tolerance Through Redundant Execution on COTS Multicores: Exploring Trade-OffsabstractHigh availability and integrity are paramount in systems deployed in life-and mission-critical scenarios. Such fault-tolerance can be achieved through redundant co-execution (RCoE) on replicated hardware, now cheaply available with multicore processors. RCoE replicates almost all software, including OS kernel, drivers, and applications, achieving a sphere of replication that covers everything except the minimal interfaces to non-replicated peripherals. We complement our original, loosely-coupled RCoE with a closely-coupled version that improves transparency of replication to application code, and investigate the functionality, performance and vulnerability trade-offs. Yanyan Shen, Gernot Heiser, Kevin Elphinstone |
DSN | 1 |
| 2019 | LADD: A Length-Adaptive Approach to Detecting Taxi Anomalous DetoursabstractThis paper focuses on the practical problem of detecting taxi anomalous detours. In this paper, we propose a length-adaptive detection approach called LADD to detect if a taxi conducted an anomalous detour by investigating the taxi journey's trajectory. We employ a similarity-based detection strategy, which takes trajectory length into consideration. Our approach can deal with both long and short trajectories by applying weighted similarity on trajectories and using trajectory concatenation or clipping. For quantitative performance comparison, we propose a performance metric, and compare our approach with the state-of-the-art methods on a real-world dataset. The results verify the superior performance of LADD on taxi detour detection. Xuansheng Lu, Weiyu Cheng, Yanyan Shen, Yanmin Zhu 0006 |
ICPADS | 3 |
| 2019 | Learning Shared Vertex Representation in Heterogeneous Graphs with Convolutional Networks for RecommendationabstractCollaborative Filtering (CF) is among the most successful techniques in recommendation tasks. Recent works have shown a boost of performance of CF when introducing the pairwise relationships between users and items or among items (users) using interaction data. However, these works usually only utilize one kind of information, i.e., user preference in a user-item interaction matrix or item dependency in interaction sequences which can limit the recommendation performance. In this paper, we propose to mine three kinds of information (user preference, item dependency, and user similarity on behaviors) by converting interaction sequence data into multiple graphs (i.e., a user-item graph, an item-item graph, and a user-subseq graph). We design a novel graph convolutional network (PGCN) to learn shared representations of users and items with the three heterogeneous graphs. In our approach, a neighbor pooling and a convolution operation are designed to aggregate features of neighbors. Extensive experiments on two real-world datasets demonstrate that our graph convolution approaches outperform various competitive methods in terms of two metrics, and the heterogeneous graphs are proved effective for improving recommendation performance. Yanmin Zhu 0006, Yanyan Shen, Jiadi Yu |
IJCAI | 3 |
| 2019 | Recurrent Neural Network for Text Classification with Hierarchical Multiscale Dense ConnectionsabstractText classification is a fundamental task in many Natural Language Processing applications. While recurrent neural networks have achieved great success in performing text classification, they fail to capture the hierarchical structure and long-term semantics dependency which are common features of text data. Inspired by the advent of the dense connection pattern in advanced convolutional neural networks, we propose a simple yet effective recurrent architecture, named Hierarchical Mutiscale Densely Connected RNNs (HM-DenseRNNs), which: 1) enables direct access to the hidden states of all preceding recurrent units via dense connections, and 2) organizes multiple densely connected recurrent units into a hierarchical multi-scale structure, where the layers are updated at different scales. HM-DenseRNNs can effectively capture long-term dependencies among words in long text data, and a dense recurrent block is further introduced to reduce the number of parameters and enhance training efficiency. We evaluate the performance of our proposed architecture on three text datasets and the results verify the advantages of HM-DenseRNN over the baseline methods in terms of the classification accuracy. Yanyan Shen |
IJCAI | 2 |
| 2019 | Incorporating Interpretability into Latent Factor Models via Fast Influence AnalysisabstractLatent factor models (LFMs) such as matrix factorization have achieved the state-of-the-art performance among various collaborative filtering approaches for recommendation. Despite the high recommendation accuracy of LFMs, a critical issue to be resolved is their lack of interpretability. Extensive efforts have been devoted to interpreting the prediction results of LFMs. However, they either rely on auxiliary information which may not be available in practice, or sacrifice recommendation accuracy for interpretability. Influence functions, stemming from robust statistics, have been developed to understand the effect of training points on the predictions of black-box models. Inspired by this, we propose a novel explanation method named FIA (Fast Influence Analysis) to understand the prediction of trained LFMs by tracing back to the training data with influence functions. We present how to employ influence functions to measure the impact of historical user-item interactions on the prediction results of LFMs and provide intuitive neighbor-style explanations based on the most influential interactions. Our proposed FIA exploits the characteristics of two important LFMs, matrix factorization and neural collaborative filtering, and is capable of accelerating the overall influence analysis process. We provide a detailed complexity analysis for FIA over LFMs and conduct extensive experiments to evaluate its performance using real-world datasets. The results demonstrate the effectiveness and efficiency of FIA, and the usefulness of the generated explanations for the recommendation results. Weiyu Cheng, Yanyan Shen, Linpeng Huang, Yanmin Zhu 0006 |
KDD | 2 |
| 2019 | Relation Extraction via Domain-aware Transfer LearningabstractRelation extraction in knowledge base construction has been researched for the last decades due to its applicability to many problems. Most classical works, such as supervised information extraction and distant supervision, focus on how to construct the knowledge base (KB) by utilizing the large number of labels or certain related KBs. However, in many real-world scenarios, the existing methods may not perform well when a new knowledge base is required but only scarce labels or few related KBs available. In this paper, we propose a novel approach called, Relation Extraction via Domain-aware Transfer Learning (ReTrans), to extract relation mentions from a given text corpus by exploring the experience from a large amount of existing KBs which may not be closely related to the target relation. We first propose to initialize the representation of relation mentions from the massive text corpus and update those representations according to existing KBs. Based on the representations of relation mentions, we investigate the contribution of each KB to the target task and propose to select useful KBs for boosting the effectiveness of the proposed approach. Based on selected KBs, we develop a novel domain-aware transfer learning framework to transfer knowledge from source domains to the target domain, aiming to infer the true relation mentions in the unstructured text corpus. Most importantly, we give the stability and generalization bound of ReTrans. Experimental results on the real world datasets well demonstrate that the effectiveness of our approach, which outperforms all the state-of-the-art baselines. Shimin Di, Yanyan Shen, Lei Chen 0002 |
KDD | 2 |
| 2019 | Attention-Based Hierarchical Recurrent Neural Network for Phenotype Classification
Yanyan Shen, Yanmin Zhu 0006 |
PAKDD (1) | 2 |
| 2019 | Learning to Fuse Multiple Semantic Aspects from Rich Texts for Stock Price Prediction
Yanyan Shen |
WISE | 2 |
| 2019 | DMDP: A Dynamic Multi-source Default Probability Prediction FrameworkabstractIn this paper, we propose a dynamic forecasting framework, named DMDP (dynamic multi-source default probability prediction), to predict the default probability of a company. The default probability is a very important factor to assess the credit risk of listed companies on a stock market. Aiming at aiding financial institutions in decision making, our DMDP framework not only analyzes financial data to capture the historical performance of a company, but also utilizes long short-term memory model to dynamically incorporate daily news from social media to take the perceptions of market participants and public opinions into consideration. The study of this paper makes two key contributions. First, we make use of unstructured news crawled from social media to alleviate the impact of financial fraud issue made on default probability prediction. Second, we propose a neural network method to integrate both structured financial factors and unstructured social media data with appropriate time alignment for default probability prediction. Extensive experimental results demonstrate the effectiveness of DMDP in predicting default probability for the listed companies in mainland China, compared with various baselines. Yanyan Shen |
Data Sci. Eng. | 2 |
| 2019 | Ensemble of 3D densely connected convolutional network for diagnosis of mild cognitive impairment and Alzheimer's disease
Yanyan Shen, Shuqiang Wang, Tengfei Xiao, Liming Deng |
Neurocomputing | 2 |
| 2019 | CATIRI: An Efficient Method for Content-and-Text Based Image Retrieval
Mengqi Zeng, Bin Yao 0002, Zhi-Jie Wang 0009, Yanyan Shen, Feifei Li 0001, Minyi Guo |
J. Comput. Sci. Technol. | 4 |
| 2019 | Fine-Grained Air Quality Inference with Remote Sensing Data and Ubiquitous Urban DataabstractAir quality has gained much attention in recent years and is of great importance to protecting people’s health. Due to the influence of multiple factors, the limited air quality monitoring stations deployed in cities are unable to provide fine-grained air quality information. One cost-effective way is to infer air quality with records from existing monitoring stations. However, the severe data sparsity problem (e.g., only 0.2% data are known) leads to the failure of most inference methods. We observe that remote sensing data are of high quality and have a strong correlation with the air quality. Therefore, we propose to integrate remote sensing data and ubiquitous urban data for the air quality inference. But there are two main challenges, i.e., data heterogeneity and incompleteness of the remote sensing data. To address the challenges, we propose a two-stage approach. In the first stage, we infer and predict air quality conditions of some places leveraging the remote sensing data and meteorological data with two proposed ANN-based methods, respectively. This stage significantly alleviates the data sparsity problem. In the second stage, the records and estimated air quality data are put in a tensor. A tensor decomposition method is applied to complete the tensor. The features extracted from urban data are classified into the spatial features (i.e., road features and POI features) and the temporal features (i.e., meteorological features) as the constraints to further address the data sparsity problem. In addition, an iterative training framework is proposed to improve the inference performance. Experiments on a real-world dataset show that our approach outperforms state-of-the-art methods, such as U-Air. Yanmin Zhu 0006, Yanyan Shen, Jiadi Yu |
ACM Trans. Knowl. Discov. Data | 3 |
| 2019 | Fast Failure Recovery in Vertex-Centric Distributed Graph Processing SystemsabstractThere is a growing need for distributed graph processing systems to have many more compute nodes processing graph-based Big Data applications, which, however, increases the chance of node failures. To address the issue, we propose a novel recovery scheme to accelerate the recovery process by parallelizing the recomputation. Once a failure occurs, all recomputations are confined to subgraphs that originally reside in the failed compute nodes. When the recovery starts, these subgraphs are reassigned to another set of compute nodes, where the recomputation over these subgraphs are conducted in parallel. To minimize the recovery latency, we also develop a reassignment strategy, from these subgraphs to the replaced compute nodes, by properly leveraging the computation and communication cost. We integrate the proposed recovery scheme into Giraph system, a widely used graph processing system. The experimental results over a variety of real graph datasets demonstrate that our proposed recovery scheme outperforms existing recovery methods by up to 30x on a cluster of 40 compute nodes. Wei Lu 0015, Yanyan Shen, Meihui Zhang 0001, H. V. Jagadish, Xiaoyong Du 0001 |
IEEE Trans. Knowl. Data Eng. | 2 |
| 2019 | Leveraging app usage contexts for app recommendation: a neural approach
Yanmin Zhu 0006, Yanyan Shen, Jiadi Yu |
World Wide Web | 3 |
| 2018 | A Neural Attention Model for Urban Air Quality Inference: Learning the Weights of Monitoring StationsabstractUrban air pollution has attracted much attention these years for its adverse impacts on human health. While monitoring stations have been established to collect pollutant statistics, the number of stations is very limited due to the high cost. Thus, inferring fine-grained urban air quality information is becoming an essential issue for both government and people. In this paper, we propose a generic neural approach, named ADAIN, for urban air quality inference. We leverage both the information from monitoring stations and urban data that are closely related to air quality, including POIs, road networks and meteorology. ADAIN combines feedforward and recurrent neural networks for modeling static and sequential features as well as capturing deep feature interactions effectively. A novel attempt of ADAIN is an attention-based pooling layer that automatically learns the weights of features from different monitoring stations, to boost the performance. We conduct experiments on a real-world air quality dataset and our approach achieves the highest performance compared with various state-of-the-art solutions. Weiyu Cheng, Yanyan Shen, Yanmin Zhu 0006, Linpeng Huang |
AAAI | 2 |
| 2018 | Stock Price Prediction Using Attention-based Multi-Input LSTMabstractStock price prediction has always been a hot but challenging task due to the complexity and randomness in stock market. Investors and researchers usually derive a great number of factors from original data such as historical stock price, company profit, or textual data collected from social media. Normally these factors are then fed into models like linear regression, SVM or neural networks to make a prediction. Even though the number of factors are considerable, most of them have relatively weak correlations with future stock price. During training process, these factors not only result in additional computation but sometimes even be harmful to the performance of prediction. In this paper, we propose a novel multi-input LSTM model which is capable of extracting valuable information from low-correlated factors and discarding their harmful noise by employing extra input gates controlled by the convincing factors called \emph{mainstream}. We also introduce several new factors including the prices of other related stocks to improve the prediction accuracy. The experimental results on the stock data from China stock market demonstrate the effectiveness of the proposed approach compared with the state-of-the-art methods. Yanyan Shen, Yanmin Zhu 0006 |
ACML | 2 |
| 2018 | A Data Driven Approach to Predicting Rating Scores for New RestaurantsabstractThis paper focuses on predicting rating scores of new restaurants listed in online restaurant review platforms. Most existing works rely on customer reviews to make an prediction. However, in practice, the customer reviews for new restaurants are always missing. In this paper, we mine useful features from the information of restaurants as well as highly available urban data to tackle this problem. We propose a deep-learning based approach called MR-Net to model both endogenous and exogenous factors in a unified manner and capture deep feature interaction for rating score prediction. Extensive experiments on real world data from Dianping show that our approach achieves better performance than various baseline methods. To the best of our knowledge, it is the first work that predicts rating scores for new restaurants without the knowledge of customer reviews. Yanyan Shen, Yanmin Zhu 0006 |
ACML | 2 |
| 2018 | A Road-Aware Neural Network for Multi-step Vehicle Trajectory Prediction
Jingze Cui, Xian Zhou 0003, Yanmin Zhu 0006, Yanyan Shen |
DASFAA (1) | 4 |
| 2018 | An Adaptive Eviction Framework for Anti-caching Based In-Memory Databases
Kaixin Huang, Shengan Zheng, Yanyan Shen, Yanmin Zhu 0006, Linpeng Huang |
DASFAA (2) | 3 |
| 2018 | Fusing Satellite Data and Urban Data for Business Location Selection: A Neural Approach
Yanyan Shen, Yanmin Zhu 0006, Jiadi Yu |
DASFAA (2) | 2 |
| 2018 | Modeling Conceptual Characteristics of Virtual Machines for CPU Utilization Prediction
Shengwei Chen, Yanyan Shen, Yanmin Zhu 0006 |
ER | 2 |
| 2018 | A GPU-Accelerated Framework for Processing Trajectory QueriesabstractThe increasing amount of trajectory data facilitates a wide spectrum of practical applications. In many such applications, large numbers of trajectory range and similarity queries are issued continuously, which calls for high-throughput trajectory query processing. Traditional in-memory databases lack considerations of the unique features of trajectories, thus suffering from inferior performance. Existing trajectory query processing systems are typically designed for only one type of trajectory queries, i.e., either range or similarity query, but not for both. Inspired by the massive parallelism on GPUs, in this paper, we develop a GPU-accelerated framework, named GAT, to support both types of trajectory queries (i.e., both range and similarity queries) with high throughput. For similarity queries, we adopt the Edit Distance on Real sequence (EDR) as the similarity measure which is accurate and robust to noise in real-world trajectories. GAT employs a GPU-friendly index called GTIDX to effectively filter invalid trajectories for both range and similarity queries, and exploits the GPU to perform parallel verifications. To accelerate the verification process on the GPU, we apply the Morton-based encoding method to reorganize trajectory points and facilitate coalesced data accesses for individual point data in global memory, which reduces the global memory bandwidth requirement significantly. We also propose a technique of grouping size-varying cells into balanced blocks with similar numbers of trajectory points, to achieve load balancing among the Streaming Multiprocessors (SMs) of the GPU. We conduct extensive experiments to evaluate the performance of GAT using two real-life trajectory datasets. The results show that GAT is scalable and achieves high throughput with acceptable indexing cost. Bowen Zhang 0006, Yanyan Shen, Yanmin Zhu 0006, Jiadi Yu |
ICDE | 2 |
| 2018 | Next Point-of-Interest Recommendation with Temporal and Multi-level Context AttentionabstractWith the prosperity of the location-based social networks, next Point-of-Interest (POI) recommendation has become an important service and received much attention in recent years. The next POI is dynamically determined by the mobility pattern and various contexts associated with user check-in sequence. However, exploring spatial-temporal mobility patterns and incorporating heterogeneous contextual factors for recommendation are challenging issues to be resolved. In this paper, we introduce a novel neural network model named TMCA (Temporal and Multi-level Context Attention) for next POI recommendation. Our model employs the LSTM-based encoder-decoder framework, which is able to automatically learn deep spatial-temporal representations for historical check-in activities and integrate multiple contextual factors using the embedding method in a unified manner. We further propose the temporal and multi-level context attention mechanisms to adaptively select relevant check-in activities and contextual factors for next POI preference prediction. Extensive experiments have been conducted using two real-world check-in datasets. The results verify (1) the superior performance of our proposed method in different evaluation metrics, compared with several state-of-the-art methods; and (2) the effectiveness of the temporal and multi-level context attention mechanisms on recommendation performance. Ranzhen Li, Yanyan Shen, Yanmin Zhu 0006 |
ICDM | 2 |
| 2018 | Forecasting Wavelet Transformed Time Series with Attentive Neural NetworksabstractThis paper studies the problem of time series forecasting. A time series is defined as a sequence of data points listed in time order. Many real-life time series data are driven by multiple latent components which occur at different frequencies. Existing solutions to time series forecasting fail to identify and discriminate these frequency-domain components. Inspired by the recent advent of signal processing and speech recognition techniques that decompose a time series signal into its time-frequency representation - a scalogram (or spectrogram), this paper proposes to explicitly disclose frequency-domain information from a univariate time series using wavelet transform, towards improving forecasting accuracy. Based on the transformed data, we leverage different neural networks to capture local time-frequency features and global long-term trend simultaneously. We further employ the attention mechanism to fuse local and global features in an effective manner. The experimental results on real time series show that our proposed approach achieves better performance than various baseline methods. Yanyan Shen, Yanmin Zhu 0006 |
ICDM | 2 |
| 2018 | Automatic Recognition of Mild Cognitive Impairment and Alzheimers Disease Using Ensemble based 3D Densely Connected Convolutional NetworksabstractAutomatic diagnosis of Alzheimers disease (AD) and mild cognition impairment (MCI) from 3D brain magnetic resonance (MR) images plays an important role in early treatment of dementia disease. Deep learning architectures can extract potential features of dementia disease and capture brain anatomical changes from MRI scans. This paper proposes an ensemble of 3D densely connected convolutional networks (3D-DenseNets) for AD and MCI diagnosis. First, dense connections were introduced to maximize the information flow, where each layer connects with all subsequent layers directly. Then weighted-based fusion method was employed to combine 3D-DenseNets with different architectures. Extensive experiments were conducted to analyze the performance of 3D-DenseNet with different hyper-parameters and architectures. Superior performance of the proposed model was demonstrated on ADNI dataset including 833 subjects. Shuqiang Wang, Yanyan Shen |
ICMLA | 3 |
| 2018 | Mining Magnitude-Oblivious Periodical Patterns of Dockless Shared Bike DemandsabstractIn spite of their convenience and popularity, dockless shared bikes pose several challenges to operations and city management, e.g., unbalanced supply and demand across a city, and harmful congestions in pedestrian walkways. A step towards addressing these challenges is to find whether there exist periodical patterns in dockless shared bike demands. A periodical pattern of dockless shared bike demand is a time-dependent demand profile that repeats over a time period and frequently appear in different areas across a city. Mining periodical patterns of dockless shared bike demands, however, is very challenging because of data noises and different demand scales in different areas. In this paper, we propose a two-stage method for mining the periodical patterns of dockless shared bike demands. In the first stage, we determine the areas which have periodical dockless shared bike demands. And in the second stage, we introduce the concept of magnitude-oblivious demand to handle the issue on the different scales in different areas and propose an algorithm to compute the magnitude-oblivious periodic patterns. We conduct experiments with real data in Shanghai, China, which show that 12.4% of spatial areas in Shanghai have periodicity in bike demands, and we report 157 magnitude-oblivious periodical patterns in Shanghai. We also present several case studies which show our approach produces meaningful and useful patterns. Interestingly, we find that the areas with a similar geography often share the same periodical patterns of dockless shared bike demands. Yanmin Zhu 0006, Yanyan Shen, Linpeng Huang |
ICPADS | 3 |
| 2018 | DELF: A Dual-Embedding based Deep Latent Factor Model for RecommendationabstractAmong various recommendation methods, latent factor models are usually considered to be state-of-the-art techniques, which aim to learn user and item embeddings for predicting user-item preferences. When applying latent factor models to recommendation with implicit feedback, the quality of embeddings always suffers from inadequate positive feedback and noisy negative feedback. Inspired by the idea of NSVD that represents users based on their interacted items, this paper proposes a dual-embedding based deep latent factor model named DELF for recommendation with implicit feedback. In addition to learning a single embedding for a user (resp. item), we represent each user (resp. item) with an additional embedding from the perspective of the interacted items (resp. users). We employ an attentive neural method to discriminate the importance of interacted users/items for dual-embedding learning. We further introduce a neural network architecture to incorporate dual embeddings for recommendation. A novel attempt of DELF is to model each user-item interaction with four deep representations that are subtly fused for preference prediction. We conducted extensive experiments on real-world datasets. The results verify the effectiveness of user/item dual embeddings and the superior performance of DELF on item recommendation. Weiyu Cheng, Yanyan Shen, Yanmin Zhu 0006, Linpeng Huang |
IJCAI | 2 |
| 2018 | Cuckoo Feature Hashing: Dynamic Weight Sharing for Sparse AnalyticsabstractFeature hashing is widely used to process large scale sparse features for learning of predictive models. Collisions inherently happen in the hashing process and hurt the model performance. In this paper, we develop a feature hashing scheme called Cuckoo Feature Hashing(CCFH) based on the principle behind Cuckoo hashing, a hashing scheme designed to resolve collisions. By providing multiple possible hash locations for each feature, CCFH prevents the collisions between predictive features by dynamically hashing them into alternative locations during model training. Experimental results on prediction tasks with hundred-millions of features demonstrate that CCFH can achieve the same level of performance by using only 15%-25% parameters compared with conventional feature hashing. Jinyang Gao, Beng Chin Ooi, Yanyan Shen, Wang-Chien Lee |
IJCAI | 3 |
| 2018 | Refine or Represent: Residual Networks with Explicit Channel-wise ConfigurationabstractThe successes of deep residual learning are mainly based on one key insight: instead of learning a completely new representation y = H(x), it is much easier to learn and optimize its residual mapping F(x)= H(x)-x, as F(x) could be generally closer to zero than the non-residual function H(x). In this paper, we further exploit this insight by explicitly configuring each feature channel with a fine-grained learning style. We define two types of channel-wise learning styles: Refine and Represent. A Refine channel is learnt via the residual function yi= Fi(x) + xi with a regularization term on the channel response ||Fi(x)||, aiming to refine the input feature channel xi of the layer. A Represent channel directly learns a new representation yi = Hi(x) without calculating the residual function with reference to xi. We apply random channel-wise configuration to each residual learning block. Experimental results on the CIFAR10, CIFAR100 and ImageNet datasets demonstrate that our proposed method can substantially improve the performance of conventional residual networks including ResNet, ResNeXt and SENet. Yanyan Shen, Jinyang Gao |
IJCAI | 1 |
| 2018 | Transfer Learning via Feature Isomorphism DiscoveryabstractTransfer learning has gained increasing attention due to the inferior performance of machine learning algorithms with insufficient training data. Most of the previous homogeneous or heterogeneous transfer learning works aim to learn a mapping function between feature spaces based on the inherent correspondence across the source and target domains or labeled instances. However, in many real world applications, existing methods may not be robust when the correspondence across domains is noisy or labeled instances are not representative. In this paper, we develop a novel transfer learning framework called Transfer Learning via Feature Isomorphism Discovery (abbreviated to TLFid), which owns high tolerance for noisy correspondence between domains as well as scarce or non-existing labeled instances. More specifically, we propose a feature isomorphism approach to discovering common substructures across feature spaces and learning a feature mapping function from the target domain to the source domain. We evaluate the performance of TLFid on the cross-lingual sentiment classification tasks. The results show that our method achieves significant improvement in terms of accuracy compared with the state-of-the-art methods. Shimin Di, Jingshu Peng, Yanyan Shen, Lei Chen 0002 |
KDD | 3 |
| 2018 | Where Will Dockless Shared Bikes be Stacked?: - Parking Hotspots Detection in a New CityabstractDockless shared bikes, which aim at providing a more flexible and convenient solution to the first-and-last mile connection, come into China and expand to other countries at a very impressing speed. The expansion of shared bike business in new cities brings many challenges among which, the most critical one is the parking chaos caused by too many bikes yet insufficient demands. To allow possible actions to be taken in advance, this paper studies the problem of detecting parking hotspots in a new city where no dockless shared bike has been deployed. We propose to measure road hotness by bike density with the help of the Kernal Density Estimation. We extract useful features from multi-source urban data and introduce a novel domain adaption network for transferring hotspots knowledge learned from one city with shared bikes to a new city. The extensive experimental results demonstrate the effectiveness of our proposed approach compared with various baselines. Zhaoyang Liu 0003, Yanyan Shen, Yanmin Zhu 0006 |
KDD | 2 |
| 2018 | Cruising or Waiting: A Shared Recommender System for Taxi Drivers
Xiaoting Jiang, Yanyan Shen, Yanmin Zhu 0006 |
PAKDD (2) | 2 |
| 2018 | Inferring Dockless Shared Bike Distribution in New CitiesabstractRecently, dockless shared bike services have achieved great success and reinvented bike sharing business in China. When expanding bike sharing business into a new city, most start-ups always wish to find out how to cover the whole city with a suitable bike distribution. In this paper, we study the problem of inferring bike distribution in new cities, which is challenging. As no dockless bikes are deployed in the new city, we propose to learn insights on bike distribution from cities populated with dockless bikes. We exploit multi-source data to identify important features that affect bike distributions and develop a novel inference model combining Factor Analysis and Convolutional Neural Network techniques. The extensive experiments on real-life datasets show that the proposed solution provides significantly more accurate inference results compared with competitive prediction methods. Zhaoyang Liu 0003, Yanyan Shen, Yanmin Zhu 0006 |
WSDM | 2 |
| 2018 | Predicting Multi-step Citywide Passenger Demands Using Attention-based Neural NetworksabstractPredicting passenger pickup/dropoff demands based on historical mobility trips has been of great importance towards better vehicle distribution for the emerging mobility-on-demand (MOD) services. Prior works focused on predicting next-step passenger demands at selected locations or hotspots. However, we argue that multi-step citywide passenger demands encapsulate both time-varying demand trends and global statuses, and hence are more beneficial to avoiding demand-service mismatching and developing effective vehicle distribution/scheduling strategies. In this paper, we propose an end-to-end deep neural network solution to the prediction task. We employ the encoder-decoder framework based on convolutional and ConvLSTM units to identify complex features that capture spatiotemporal influences and pickup-dropoff interactions on citywide passenger demands. A novel attention model is incorporated to emphasize the effects of latent citywide mobility regularities. We evaluate our proposed method using real-word mobility trips (taxis and bikes) and the experimental results show that our method achieves higher prediction accuracy than the adaptations of the state-of-the-art approaches. Xian Zhou 0003, Yanyan Shen, Yanmin Zhu 0006, Linpeng Huang |
WSDM | 2 |
| 2018 | Classification of Diffusion Tensor Metrics for the Diagnosis of a Myelopathic Cord Using Machine LearningabstractIn this study, we propose an automated framework that combines diffusion tensor imaging (DTI) metrics with machine learning algorithms to accurately classify control groups and groups with cervical spondylotic myelopathy (CSM) in the spinal cord. The comparison between selected voxel-based classification and mean value-based classification were performed. A support vector machine (SVM) classifier using a selected voxel-based dataset produced an accuracy of 95.73%, sensitivity of 93.41% and specificity of 98.64%. The efficacy of each index of diffusion for classification was also evaluated. Using the proposed approach, myelopathic areas in CSM are detected to provide an accurate reference to assist spine surgeons in surgical planning in complicated cases. Shuqiang Wang, Yong Hu 0003, Yanyan Shen, Han-Xiong Li |
Int. J. Neural Syst. | 3 |
| 2018 | SMOPAT: Mining semantic mobility patterns from trajectories of private vehicles
Chengcheng Wan 0001, Yanmin Zhu 0006, Jiadi Yu, Yanyan Shen |
Inf. Sci. | 4 |
| 2018 | NVHT: An efficient key-value storage library for non-volatile memory
Kaixin Huang, Linpeng Huang, Yanyan Shen |
J. Parallel Distributed Comput. | 4 |
| 2018 | HMFS: A hybrid in-memory file system with version consistency
Hao Liu 0085, Linpeng Huang, Yanmin Zhu 0006, Shengan Zheng, Yanyan Shen |
J. Parallel Distributed Comput. | 5 |
| 2018 | HMVFS: A Versioning File System on DRAM/NVM Hybrid Memory
Shengan Zheng, Hao Liu 0085, Linpeng Huang, Yanyan Shen, Yanmin Zhu 0006 |
J. Parallel Distributed Comput. | 4 |
| 2018 | Top-kCritical Vertices Query on Shortest PathabstractShortest path query is one of the most fundamental and classic problems in graph analytics, which returns the complete shortest path between any two vertices. However, in many real-life scenarios, only critical vertices on the shortest path are desirable and it is unnecessary to search for the complete path. This paper investigates the shortest path sketch by defining a top- $k$ critical vertices ( $k$ CV) query on the shortest path. Given a source vertex $s$ and target vertex $t$ in a graph, $k$ CV query can return the top- $k$ significant vertices on the shortest path $SP(s,t)$ . The significance of the vertices can be predefined. The key strategy for seeking the sketch is to apply off-line preprocessed distance oracle to accelerate on-line real-time queries. This allows us to omit unnecessary vertices and obtain the most representative sketch of the shortest path directly. We further explore a series of methods and optimizations to answer $k$ CV query on both centralized and distributed platforms, using exact and approximate approaches, respectively. We evaluate our methods in terms of time, space complexity and approximation quality. Experiments on large-scale real-world networks validate that our algorithms are of high efficiency and accuracy. Jing Ma 0002, Bin Yao 0002, Xiaofeng Gao 0001, Yanyan Shen, Minyi Guo |
IEEE Trans. Knowl. Data Eng. | 4 |
| 2017 | Automatic Recognition of Mild Cognitive Impairment from MRI Images Using Expedited Convolutional Neural Networks
Shuqiang Wang, Yanyan Shen, Tengfei Xiao, Jinxing Hu |
ICANN (1) | 2 |
| 2017 | Adaptive Prefetching for Accelerating Read and Write in NVM-Based File SystemsabstractThe byte-addressable Non-Volatile Memory (NVM) offers fast, fine-grained access to persistent storage. While DRAM and NVM have similar read performance, the write operations of existing NVM materials incur longer latency and lower bandwidth than DRAM. This read-write asymmetry nature of NVM causes two bottlenecks for accessing read-and write-intensive file data: expensive data block lookups via file inner structure and high-latency direct writes to data blocks in NVM. However, existing NVM-based file systems fail to address both bottlenecks well. This paper presents WARP, an adaptive prefetching module designed for NVM-based file systems, which aims to deal with two bottlenecks effectively. WARP employs two acceleration approaches: 1) mapping data blocks into kernel virtual address space to bypass the indirection of file inner structure for read-intensive file data; and 2) allocating DRAM buffer to absorb frequent writes for write-intensive file data. We design a WARP benefit model to identify read-and write-intensive access patterns for file data, and use a successor prediction model to predict future data access based on historical file access traces. With WARP, we are able to prefetch file data according to both file access patterns and traces with consistency guarantee. WARP can be implemented on various NVM-based file systems, and we choose HMVFS for the experiments. The evaluation results show that HMVFS with WARP provides high prefetching accuracy and up to 32%-83% improvement compared with the state-of-the-art NVM-based file systems. Shengan Zheng, Hong Mei 0001, Linpeng Huang, Yanyan Shen, Yanmin Zhu 0006 |
ICCD | 4 |
| 2017 | A DAX-enabled mmap mechanism for log-structured in-memory file systemsabstractEmerging byte-addressable Non-Volatile Memory(NVM) technologies offer fine-grained access to persistent data with latency comparable to DRAM. This presents both challenges and opportunities to system software designers. To utilize the fascinating features of NVM, a lot of works have been done to develop NVM-based file systems. Direct access (DAX) is a key feature provided by most NVM-based file systems, it enables applications to directly access NVM without going through the page cache layer. However, DAX-style mmap may cause serious consistency issues in versioning in-memory file systems that adopt CoW to solve block sharing problem among snapshots. In this paper, we propose a new mmap mechanism called versioning-mmap which solves the consistency issue of memory-mapped I/O in DAX-enabled versioning in-memory file systems. We implement our proposed mechanism in HMVFS and do a series of experiments based on our implementation. Results show that our mechanism achieves consistency while imposing negligible performance overhead compared to PMFS and NOVA. Zhixiang Mao, Shengan Zheng, Linpeng Huang, Yanyan Shen |
IPCCC | 4 |
| 2017 | Fair Resource Allocation Algorithm for Chunk Based OFDMA Multi-User NetworksabstractThis paper investigates the resource allocation problem in orthogonal frequency division multiple access multi-user networks, where subcarriers are grouped into chunks due to simplicity of implementation. The aim is to achieve max-min fairness among users by adjusting the transmission power allocation and chunk allocation while taking into account several important constrains. The problem is formulated as a mixed integer nonlinear programming problem, whose optimal solution is extremely hard to find. Then a low complexity suboptimal algorithm is proposed, which solves the chunk allocation and power allocation in two steps separately. A fast optimal power allocation algorithm is designed by exploiting the special structure of the problem. Simulations verify the performance of the proposed algorithm in terms of the users' minimal transmission rate and running time comparing with benchmark algorithms. Yanyan Shen, Xiaoxia Huang 0004, Bo Yang 0006, Shimin Gong, Shuqiang Wang |
VTC Fall | 1 |
| 2016 | NVHT: an efficient key-value storage library for non-volatile memoryabstractModern Non-Volatile Memory (NVM) promises persistence, byte-addressability and DRAM-like read and write latency, which offers great opportunities for big data storage architecture. These excellent properties indicate that NVM has the potential to be incorporated with key-value stores to achieve high performance and durability simultaneously. Yanyan Shen, Sumin Li, Linpeng Huang |
BDCAT | 2 |
| 2016 | Robust Relay Beamforming in Device-to-Device Networks with Energy Harvesting ConstraintsabstractMotivated by the observation that energy harvesting (EH) from radio-frequency (RF) signal is subject to fluctuations, multiple EH-enabled relays are employed to collaboratively enhance data communications in a device-to-device (D2D) network underlying a cellular system. Each relay is equipped with a single antenna and unable to harvest energy and transmit data simultaneously. Thus, the D2D user equipment (DUE) needs to optimally schedule the channel time for the relays' EH and data transmissions, which depends on their EH capabilities and channel conditions. Considering that the relays' channel estimations are usually unreliable, we formulate a robust throughput maximization problem to optimize the relays' EH time and transmit power, subject to a probabilistic interference constraint at the cellular user equipment (CUE). We show that the proposed problem, though non-convex, can be tackled by exploiting its monotonicity structure. Specifically, we design a successive approximation algorithm that involves solving a sequence of semi-definite programs (SDPs) and show numerically that it always achieves the global optimum. This validates our analysis and demonstrates the efficacy of the proposed algorithm. Shimin Gong, Yanyan Shen, Xiaoxia Huang 0004, Sissi Xiaoxiao Wu, Anthony Man-Cho So |
GLOBECOM | 2 |
| 2015 | Resource Allocation for OFDMA Relay Networks with Wireless Information and Power TransferabstractIn this paper, we investigate the resource allocation for orthogonal frequency division multiple access relay networks, where the relay does not have embedded energy supply and needs to first harvest energy from the received signals from the source before forwarding transmission. The relay uses time switching scheme for wireless information and power transfer. We aim to maximize the weighted sum rate under several constraints by varying the source transmission power, the relay transmission power, and the time switching ratio. We formulate the joint resource allocation problem as an optimization problem, which is non-convex. Although it is difficult to solve the non-convex problem, we derive its closed-form solution by exploiting its special structure. We also prove that the closed- form solution is a partial optimum. Finally, simulations verify the proposed closed-form solution is superior to the equal power solution. Yanyan Shen, Kyung Sup Kwak, Bo Yang 0006, Shuqiang Wang, Xiaoxia Huang 0004, Xin-Ping Guan, Ramesh R. Rao |
GLOBECOM | 1 |
| 2015 | Hadoop-Based Analysis for Large-Scale Click-Through Patterns in 4G Network
Shuqiang Wang, Yanyan Shen, Jinxing Hu, Zhe Xuan, Zhe Lu |
WASA | 2 |
| 2015 | DEXTER: Large-Scale Discovery and Extraction of Product Specifications on the WebabstractThe web is a rich resource of structured data. There has been an increasing interest in using web structured data for many applications such as data integration, web search and question answering. In this paper, we present Dexter, a system to find product sites on the web, and detect and extract product specifications from them. Since product specifications exist in multiple product sites, our focused crawler relies on search queries and backlinks to discover product sites. To perform the detection, and handle the high diversity of specifications in terms of content, size and format, our system uses supervised learning to classify HTML fragments (e.g., tables and lists) present in web pages as specifications or not. To perform large-scale extraction of the attribute-value pairs from the HTML fragments identified by the specification detector, D exter adopts two lightweight strategies: a domain-independent and unsupervised wrapper method, which relies on the observation that these HTML fragments have very similar structure; and a combination of this strategy with a previous approach, which infers extraction patterns by annotations generated by automatic but noisy annotators. The results show that our crawler strategy to locate product specification pages is effective: (1) it discovered 1:46A M product specification pages from 3; 005 sites and 9 different categories; (2) the specification detector obtains high values of F-measure (close to 0:9) over a heterogeneous set of product specifications; and (3) our efficient wrapper methods for attribute-value extraction get very high values of precision (0.92) and recall (0.95) and obtain better results than a state-of-the-art, supervised rule-based wrapper. Disheng Qiu, Luciano Barbosa, Xin Dong 0001, Yanyan Shen, Divesh Srivastava |
Proc. VLDB Endow. | 4 |
| 2015 | Opportunistic multichannel access with decentralized channel state informationabstractThis paper considers multiaccess control for the uplink in orthogonal frequency division multiple access wireless networks. To avoid the extensive information exchange associated with centralized approaches, we formulate the decentralized access control problem with the contention power constraint as a Bayesian game, mapping time-varying channel state information into contention strategies. By exploiting the problem structure, a strategy where users access the channels with probability one if the observed channel gain is above a predetermined threshold is shown to be optimal. It is also shown that the energy consumption of the threshold strategy will not exceed that of randomized strategies. The game is then equivalently reformulated as one of finding the threshold value in a distributed manner, and the existence and uniqueness of Bayesian Nash equilibria is established. A distributed algorithm based on Lagrange duality is proposed to approach the unique equilibrium, and the algorithm is shown to be globally stable. In a homogeneous system, the performance loss of the proposed scheme is proved to be bounded compared with a centralized channel allocation scheme. Contrary to other proposals, our method allows for heterogeneous channel state information and achieves a comparable throughput with reduced power. Copyright © 2013 John Wiley & Sons, Ltd. Bo Yang 0006, Yanyan Shen, Mikael Johansson 0001, Cailian Chen, Xin-Ping Guan |
Wirel. Commun. Mob. Comput. | 2 |
| 2014 | Discovering queries based on example tuplesabstractAn enterprise information worker is often aware of a few example tuples (but not the entire result) that should be present in the output of the query. We study the problem of discovering the minimal project join query that contains the given example tuples in its output. Efficient discovery of such queries is challenging. We propose novel algorithms to solve this problem. Our experiments on real-life datasets show that the proposed solution is significantly more efficient compared with na\"{i}ve adaptations of known techniques. Yanyan Shen, Kaushik Chakrabarti, Surajit Chaudhuri, Bolin Ding, Lev Novik |
SIGMOD Conference | 1 |
| 2014 | Fast Failure Recovery in Distributed Graph Processing SystemsabstractDistributed graph processing systems increasingly require many compute nodes to cope with the requirements imposed by contemporary graph-based Big Data applications. However, increasing the number of compute nodes increases the chance of node failures. Therefore, provisioning an efficient failure recovery strategy is critical for distributed graph processing systems. This paper proposes a novel recovery mechanism for distributed graph processing systems that parallelizes the recovery process. The key idea is to partition the part of the graph that is lost during a failure among a subset of the remaining nodes. To do so, we augment the existing checkpoint-based and log-based recovery schemes with a partitioning mechanism that is sensitive to the total computation and communication cost of the recovery process. Our implementation on top of the widely used Giraph system outperforms checkpoint-based recovery by up to 30x on a cluster of 40 compute nodes. Yanyan Shen, Gang Chen 0001, H. V. Jagadish, Wei Lu 0015, Beng Chin Ooi, Bogdan Marius Tudor |
Proc. VLDB Endow. | 1 |
| 2014 | Resource allocation with proportional rate fairness in orthogonal frequency division multiple access relay networksabstractWe address the problem of subchannel and transmission power allocation in orthogonal frequency division multiple access relay networks with an aim to maximize the sum rate and maintain proportional rate fairness among users. Because the formulated problem is a mixed-integer nonlinear optimization problem with an extremely high computational complexity, we propose a low-complexity suboptimal algorithm, which is a two-step separated subchannel and power allocation algorithm. In the first step, subchannels are allocated to each user, whereas in the second step, the optimal power allocation is carried out on the basis of the given subchannel allocation and the nonlinear interval Gauss–Seidel method. Simulation results have demonstrated that the proposed algorithm can achieve a good trade-off between the efficiency and the fairness compared with two other existing relevant algorithms. In particular, the proposed algorithm can always achieve 100% fairness under various conditions. Copyright © 2012 John Wiley & Sons, Ltd. Yanyan Shen, Gang Feng 0001, Bo Yang 0006, Xin-Ping Guan |
Wirel. Commun. Mob. Comput. | 1 |
| 2013 | Increasing the trustworthiness of commodity hardware through softwareabstractAdvances in formal software verification has produced an operating system that is guaranteed mathematically to be correct and enforce access isolation. Such an operating system could potentially consolidate safety and security critical software on a single device where previously multiple devices were used. One of the barriers to consolidation on commodity hardware is the lack of hardware dependability features. A hardware fault triggered by cosmic rays, alpha particle strikes, etc. potentially invalidates the strong mathematical guarantees. This paper discusses improving the trustworthiness of commodity hardware to enable a verified microkernel to be used in some situations previously needing separate computers. We explore leveraging multicore processors to provide redundancy, and report the results of our initial performance investigation. Kevin Elphinstone, Yanyan Shen |
DSN | 2 |
| 2012 | CDAS: A Crowdsourcing Data Analytics SystemabstractSome complex problems, such as image tagging and natural language processing, are very challenging for computers, where even state-of-the-art technology is yet able to provide satisfactory accuracy. Therefore, rather than relying solely on developing new and better algorithms to handle such tasks, we look to the crowdsourcing solution -- employing human participation -- to make good the shortfall in current technology. Crowdsourcing is a good supplement to many computer tasks. A complex job may be divided into computer-oriented tasks and human-oriented tasks, which are then assigned to machines and humans respectively. To leverage the power of crowdsourcing, we design and implement a Crowdsourcing Data Analytics System, CDAS. CDAS is a framework designed to support the deployment of various crowdsourcing applications. The core part of CDAS is a quality-sensitive answering model, which guides the crowdsourcing engine to process and monitor the human tasks. In this paper, we introduce the principles of our quality-sensitive model. To satisfy user required accuracy, the model guides the crowdsourcing query engine for the design and processing of the corresponding crowdsourcing jobs. It provides an estimated accuracy for each generated result based on the human workers' historical performances. When verifying the quality of the result, the model employs an online strategy to reduce waiting time. To show the effectiveness of the model, we implement and deploy two analytics jobs on CDAS, a twitter sentiment analytics job and an image tagging job. We use real Twitter and Flickr data as our queries respectively. We compare our approaches with state-of-the-art classification and image annotation techniques. The results show that the human-assisted methods can indeed achieve a much higher accuracy. By embedding the quality-sensitive model into crowdsourcing query engine, we effectively reduce the processing cost while maintaining the required query answer quality. Meiyu Lu, Beng Chin Ooi, Yanyan Shen, Sai Wu, Meihui Zhang 0001 |
Proc. VLDB Endow. | 4 |
| 2012 | Efficient Processing of k Nearest Neighbor Joins using MapReduceabstractk nearest neighbor join ( k NN join), designed to find k nearest neighbors from a dataset S for every object in another dataset R , is a primitive operation widely adopted by many data mining applications. As a combination of the k nearest neighbor query and the join operation, k NN join is an expensive operation. Given the increasing volume of data, it is difficult to perform a k NN join on a centralized machine efficiently. In this paper, we investigate how to perform k NN join using MapReduce which is a well-accepted framework for data-intensive applications over clusters of computers. In brief, the mappers cluster objects into groups; the reducers perform the k NN join on each group of objects separately. We design an effective mapping mechanism that exploits pruning rules for distance filtering, and hence reduces both the shuffling and computational costs. To reduce the shuffling cost, we propose two approximate algorithms to minimize the number of replicas. Extensive experiments on our in-house cluster demonstrate that our proposed methods are efficient, robust and scalable. Wei Lu 0015, Yanyan Shen, Beng Chin Ooi |
Proc. VLDB Endow. | 2 |
| 2010 | Threshold-Based Multichannel Access with Energy ConstraintabstractThis paper considers multiaccess control for the uplink in orthogonal-frequency-division-multiple-access (OFDMA) wireless networks. To avoid extensive information exchange with the access point in centralized approaches, we propose a distributed threshold-based scheme, where each user accesses multiple channels simultaneously based on a comparison between measured channel gains and a channel gain threshold. Each user will adapts its channel gain threshold based on local measurements of collision on each channel and the energy consumption for channel contention. The problem is formulated as a constrained non-cooperative game. We show existence and uniqueness of the Nash equilibrium. A gradient-based algorithm is proposed to update the channel gain threshold. Furthermore, the convergence of this algorithm is proved. In addition, for heterogeneous systems, our proposed scheme can maintain multiuser diversity gains considering the time-varying channel gain and energy consumption. Compared with peer distributed OFDMA schemes and random channel selection algorithms, our proposed schemes reduce overhead and achieve a higher throughput. Bo Yang 0006, Yanyan Shen, Mikael Johansson 0001, Xin-Ping Guan |
ICC | 2 |
| 2009 | Delay-dependent stability and Hinfinity control for a class of fuzzy descriptor systems with time-delay
Hongbin Zhang 0002, Yanyan Shen, Gang Feng 0001 |
Fuzzy Sets Syst. | 2 |
| 2008 | Distributed power control and random access for spectrum sharing with QoS constraint
Bo Yang 0006, Yanyan Shen, Gang Feng 0001, Chengnian Long, Zhong-Ping Jiang, Xin-Ping Guan |
Comput. Commun. | 2 |