Demo Data from SIGIR 2026 / reviewers in the wild / expert
Yujing Wang 0002
dblp:16/4075-2
· DBLP profile ↗
32ranked-venue papers
6as first author
23since 2021 · last 2025
0000-0002-7940-5216ORCID · conflict
Domains — the database's venue-derived domains; a paper can count in several
Artificial intelligence and machine learning · 31 · 6 first-author · 22 since 2021Databases, data management, data science and information retrieval · 11 · 1 first-author · 8 since 2021Graphics, computer vision, multimedia, augmented reality and games · 9 · 2 first-author · 6 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
17 papers |
Efficient and distributed learning · 27% Deep learning architectures and training · 15% Graph learning · 11% | |
| Databases, data mining, and information retrieval
10 papers |
Information retrieval · 52% Data mining · 19% Recommender systems · 12% |
Topics — the 30 heaviest of 61, each with the papers that count most for it
| Topic | Weight | Papers | Last | Evidence papers |
|---|---|---|---|---|
Machine learning › Efficient and distributed learning
model compression |
1.3 | 2 | 2023 | Constraint-aware and Ranking-distilled Token Pruning for Efficient Transformer Inference · KDD 2023 SpaceEvo: Hardware-Friendly Search Space Design for Efficient INT8 Inference · ICCV 2023 |
Information retrieval
retrieval models |
1.2 | 2 | 2023 | Model-enhanced Vector Index · NeurIPS 2023 A Neural Corpus Indexer for Document Retrieval · NeurIPS 2022 |
Machine learning › Deep learning architectures and training
attention mechanism |
1.2 | 2 | 2023 | Convolution-Enhanced Evolving Attention Networks · IEEE Trans. Pattern Anal. Mach. Intell. 2023 Evolving Attention with Residual Convolutions · ICML 2021 |
Machine learning › Efficient and distributed learning › automated machine learning
neural architecture search |
1.1 | 2 | 2023 | SpaceEvo: Hardware-Friendly Search Space Design for Efficient INT8 Inference · ICCV 2023 TextNAS: A Neural Architecture Search Space Tailored for Text Representation · AAAI 2020 |
Machine learning › Graph learning
graph neural network |
0.9 | 2 | 2021 | DeGNN: Improving Graph Neural Networks with Graph Decomposition · KDD 2021 Spectral Temporal Graph Neural Network for Multivariate Time-series Forecasting · NeurIPS 2020 |
Machine learning › Deep learning architectures and training
convolutional neural network |
0.9 | 2 | 2021 | Evolving Attention with Residual Convolutions · ICML 2021 Time-Series Anomaly Detection Service at Microsoft · KDD 2019 |
Machine learning › Efficient and distributed learning › parameter-efficient fine-tuning
low-rank adaptation |
0.9 | 1 | 2025 | MTL-LoRA: Low-Rank Adaptation for Multi-Task Learning · AAAI 2025 |
Machine learning › Learning paradigms
multi-task learning |
0.9 | 1 | 2025 | MTL-LoRA: Low-Rank Adaptation for Multi-Task Learning · AAAI 2025 |
Machine learning › Efficient and distributed learning
parameter-efficient fine-tuning |
0.9 | 1 | 2025 | MTL-LoRA: Low-Rank Adaptation for Multi-Task Learning · AAAI 2025 |
Machine learning › Learning paradigms › multi-task learning
task conflict mitigation |
0.9 | 1 | 2025 | MTL-LoRA: Low-Rank Adaptation for Multi-Task Learning · AAAI 2025 |
Data mining
anomaly detection |
0.8 | 2 | 2020 | Multivariate Time-series Anomaly Detection via Graph Attention Network · ICDM 2020 Time-Series Anomaly Detection Service at Microsoft · KDD 2019 |
Information retrieval
image retrieval |
0.8 | 1 | 2024 | IRGen: Generative Modeling for Image Retrieval · ECCV (15) 2024 |
Machine learning › Probabilistic and Bayesian machine learning
causal inference |
0.7 | 1 | 2023 | Estimating Treatment Effects from Irregular Time Series Observations with Hidden Confounders · AAAI 2023 |
Machine learning › Efficient and distributed learning › automated machine learning › neural architecture search
hardware-aware neural architecture search |
0.7 | 1 | 2023 | SpaceEvo: Hardware-Friendly Search Space Design for Efficient INT8 Inference · ICCV 2023 |
Machine learning › Probabilistic and Bayesian machine learning › causal inference › causal effect estimation › treatment effect estimation
individual treatment effect estimation |
0.7 | 1 | 2023 | Estimating Treatment Effects from Irregular Time Series Observations with Hidden Confounders · AAAI 2023 |
Machine learning › Deep learning architectures and training › neural differential equations
neural controlled differential equations |
0.7 | 1 | 2023 | Estimating Treatment Effects from Irregular Time Series Observations with Hidden Confounders · AAAI 2023 |
Natural language and speech › Language models and text generation › prompting
prompt engineering |
0.7 | 1 | 2023 | UPRISE: Universal Prompt Retrieval for Improving Zero-Shot Evaluation · EMNLP 2023 |
Natural language and speech › Language models and text generation › prompting
prompt search |
0.7 | 1 | 2023 | UPRISE: Universal Prompt Retrieval for Improving Zero-Shot Evaluation · EMNLP 2023 |
Machine learning › Efficient and distributed learning › model compression
quantization |
0.7 | 1 | 2023 | SpaceEvo: Hardware-Friendly Search Space Design for Efficient INT8 Inference · ICCV 2023 |
Machine learning › Representation and self-supervised learning › representation learning › sequence representation learning
time series representation learning |
0.7 | 1 | 2023 | Estimating Treatment Effects from Irregular Time Series Observations with Hidden Confounders · AAAI 2023 |
Machine learning › Efficient and distributed learning › model compression
token pruning |
0.7 | 1 | 2023 | Constraint-aware and Ranking-distilled Token Pruning for Efficient Transformer Inference · KDD 2023 |
Computer vision › Vision and language
vision-language pretraining |
0.7 | 1 | 2023 | VL-Match: Enhancing Vision-Language Pretraining with Token-Level and Instance-Level Matching · ICCV 2023 |
Machine learning › Transfer learning and domain adaptation
zero-shot evaluation |
0.7 | 1 | 2023 | UPRISE: Universal Prompt Retrieval for Improving Zero-Shot Evaluation · EMNLP 2023 |
Information retrieval
cross-modal retrieval |
0.7 | 1 | 2023 | VL-Match: Enhancing Vision-Language Pretraining with Token-Level and Instance-Level Matching · ICCV 2023 |
Information retrieval › retrieval models › neural retrieval
dense retrieval |
0.7 | 1 | 2023 | Model-enhanced Vector Index · NeurIPS 2023 |
Information retrieval › cross-modal retrieval
image-text retrieval |
0.7 | 1 | 2023 | VL-Match: Enhancing Vision-Language Pretraining with Token-Level and Instance-Level Matching · ICCV 2023 |
Knowledge graphs
link prediction |
0.7 | 1 | 2023 | To Copy Rather Than Memorize: A Vertical Learning Paradigm for Knowledge Graph Completion · ACL (1) 2023 |
Indexing and storage engines
vector index |
0.7 | 1 | 2023 | Model-enhanced Vector Index · NeurIPS 2023 |
Information retrieval
document retrieval |
0.6 | 1 | 2022 | A Neural Corpus Indexer for Document Retrieval · NeurIPS 2022 |
Recommender systems
session-based recommendation |
0.6 | 1 | 2022 | Learning Multi-granularity Consecutive User Intent Unit for Session-based Recommendation · WSDM 2022 |
Methods — techniques the papers use, named apart from their topics
generative modeling · 1.5replaced token detection · 1.3quantization-friendly search space design · 1.3hard negative mining · 1.3bootstrapping · 1.3convolutional neural network · 1.0low-rank adaptation · 0.9twin-tower model · 0.7residual quantization · 0.7quantized-for-all supernet · 0.7prompt retrieval · 0.7neural controlled differential equations · 0.7lipschitz regularization · 0.7copy mechanism · 0.7sequence-to-sequence network · 0.6graph neural network · 0.6consistency regularization · 0.6
| Year | Publication | Venue | Position |
|---|---|---|---|
| 2025 | MTL-LoRA: Low-Rank Adaptation for Multi-Task LearningabstractParameter-efficient fine-tuning (PEFT) has been widely employed for domain adaptation, with LoRA being one of the most prominent methods due to its simplicity and effectiveness. However, in multi-task learning (MTL) scenarios, LoRA tends to obscure the distinction between tasks by projecting sparse high-dimensional features from different tasks into the same dense low-dimensional intrinsic space. This leads to task interference and suboptimal performance for LoRA and its variants. To tackle this challenge, we propose MTL-LoRA, which retains the advantages of low-rank adaptation while significantly enhancing MTL capabilities. MTL-LoRA augments LoRA by incorporating additional task-adaptive parameters that differentiate task-specific information and capture shared knowledge across various tasks within low-dimensional spaces. This approach enables pretrained models to jointly adapt to different target domains with a limited number of trainable parameters. Comprehensive experimental results, including evaluations on public academic benchmarks for natural language understanding, commonsense reasoning, and image-text understanding, as well as real-world industrial text Ads relevance datasets, demonstrate that MTL-LoRA outperforms LoRA and its various variants with comparable or even fewer learnable parameters in MTL setting. Yaming Yang 0001, Dilxat Muhtar, Yelong Shen, Yuefeng Zhan, Yujing Wang 0002, Hao Sun 0015, Feng Sun 0008, Qi Zhang 0066, Weizhu Chen, Yunhai Tong |
AAAI | 6 |
| 2025 | CSRM-LLM: Embracing Multilingual LLMs for Cold-Start Relevance Matching in Emerging E-commerce MarketsabstractAs global e-commerce platforms continue to expand, companies are entering new markets where they encounter cold-start challenges due to limited human labels and user behaviors. In this paper, we share our experiences in Coupang to provide a competitive cold-start performance of relevance matching for emerging e-commerce markets. Specifically, we present a Cold-Start Relevance Matching (CSRM) framework, utilizing a multilingual Large Language Model (LLM) to address three challenges: (1) activating cross-lingual transfer learning abilities of LLMs through machine translation tasks; (2) enhancing query understanding and incorporating e-commerce knowledge by retrieval-based query augmentation; (3) mitigating the impact of training label errors through a multi-round self-distillation training strategy. Our experiments demonstrate the effectiveness of CSRM-LLM and the proposed techniques, resulting in successful real-world deployment and significant online gains, with a 45.8% reduction in defect ratio and a 0.866% uplift in session purchase rate. Yujing Wang 0002, Huoran Li, Chunxu Xu, Yuchong Luo, Xianghui Mao, Cong Li 0021, Lun Du, Chunyang Ma, Qiqi Jiang, Wenting Mo, Pei Wen, Shantanu Kumar, Taejin Park, Yiwei Song, Vijay Rajaram, Sonu Durgia, Pranam Kolari |
CIKM | 1 |
| 2024 | You Can't Ignore Either: Unifying Structure and Feature Denoising for Robust Graph LearningabstractRecent research on the robustness of Graph Neural Networks (GNNs) under noises or attacks has attracted great attention due to its importance in real-world applications. Most previous methods explore a single noise source, recovering corrupt node embedding by reliable structures bias or developing structure learning with reliable node features. However, the noises and attacks may come from both structures and features in graphs, making the graph denoising a dilemma and challenging problem. In this paper, we develop a unified graph denoising (UGD) framework to unravel the deadlock between structure and feature denoising. Specifically, a high-order neighborhood proximity evaluation method is proposed to recognize noisy edges, considering features may be perturbed simultaneously. Moreover, we propose to refine noisy features with reconstruction based on a graph auto-encoder. An iterative updating algorithm is further designed to optimize the framework and acquire a clean graph, thus enabling robust graph learning for downstream tasks. Our UGD framework is self-supervised and can be easily implemented as a plug-and-play module. We carry out extensive experiments, which proves the effectiveness and advantages of our method. Code is avalaible at https://github.com/YoungTimmy/UGD. Tianmeng Yang, Jiahao Meng, Min Zhou 0006, Yaming Yang 0001, Yujing Wang 0002, Xiangtai Li, Yunhai Tong |
CIKM | 5 |
| 2024 | IRGen: Generative Modeling for Image Retrieval
Ting Zhang 0002, Dong Chen 0003, Yujing Wang 0002, Qi Chen 0009, Xing Xie 0001, Hao Sun 0015, Qi Zhang 0066, Fan Yang 0024, Mao Yang 0004, Qingmin Liao, Jingdong Wang 0001, Baining Guo |
ECCV (15) | 4 |
| 2024 | Collaborative Multi-Task Representation for Natural Language UnderstandingabstractMulti-task learning has shown large benefits in Natural Language Understanding (NLU). However, current state-of-the-arts (SOTAs) like MT-DNN and MMoE do not model task relationships explicitly and fail to obtain effective task alignment. In this paper, we propose a Collaborative Multi-Task Representation (CMTR) framework to tackle this problem. We capture instance-level task relations through a task interaction layer, which helps guide the fusion of task-oriented representations into the final representation. Moreover, tailored loss functions are proposed to facilitate the learning of task alignment. Specifically, we leverage knowledge distillation as an auxiliary loss to assist the adaptation layers in generating task-oriented representations. We also introduce a regularization loss to learn better gating functions for multi-task fusion. Empirically, CMTR outperforms SOTA multi-task learning frameworks on most natural language understanding tasks in the GLUE benchmark. Furthermore, it achieves better task alignment and demonstrates good interpretability. Yaming Yang 0001, Defu Cao, Ming Zeng 0009, Jing Yu 0007, Yunhai Tong, Yujing Wang 0002 |
IJCNN | 7 |
| 2024 | Customizing graph neural networks using path reweighting
Jianpeng Chen, Yujing Wang 0002, Ming Zeng 0009, Zongyi Xiang, Bitan Hou, Tong Yu 0001, Ole J. Mengshoel, Yazhou Ren 0001 |
Inf. Sci. | 2 |
| 2023 | Estimating Treatment Effects from Irregular Time Series Observations with Hidden ConfoundersabstractCausal analysis for time series data, in particular estimating individualized treatment effect (ITE), is a key task in many real world applications, such as finance, retail, healthcare, etc. Real world time series, i.e., large-scale irregular or sparse and intermittent time series, raise significant challenges to existing work attempting to estimate treatment effects. Specifically, the existence of hidden confounders can lead to biased treatment estimates and complicate the causal inference process. In particular, anomaly hidden confounders which exceed the typical range can lead to high variance estimates. Moreover, in continuous time settings with irregular samples, it is challenging to directly handle the dynamics of causality. In this paper, we leverage recent advances in Lipschitz regularization and neural controlled differential equations (CDE) to develop an effective and scalable solution, namely LipCDE, to address the above challenges. LipCDE can directly model the dynamic causal relationships between historical data and outcomes with irregular samples by considering the boundary of hidden confounders given by Lipschitz constrained neural networks. Furthermore, we conduct extensive experiments on both synthetic and real world datasets to demonstrate the effectiveness and scalability of LipCDE. Defu Cao, James Enouen, Yujing Wang 0002, Xiangchen Song, Chuizheng Meng, Hao Niu 0001, Yan Liu 0002 |
AAAI | 3 |
| 2023 | To Copy Rather Than Memorize: A Vertical Learning Paradigm for Knowledge Graph CompletionabstractRui Li, Xu Chen, Chaozhuo Li, Yanming Shen, Jianan Zhao, Yujing Wang, Weihao Han, Hao Sun, Weiwei Deng, Qi Zhang, Xing Xie. Proceedings of the 61st Annual Meeting of the Association for Computational Linguistics (Volume 1: Long Papers). 2023. Rui Li 0086, Xu Chen 0017, Chaozhuo Li, Yanming Shen, Jianan Zhao 0002, Yujing Wang 0002, Weihao Han, Hao Sun 0015, Qi Zhang 0066, Xing Xie 0001 |
ACL (1) | 6 |
| 2023 | Mitigating Semantic Confusion from Hostile Neighborhood for Graph Active LearningabstractGraph Active Learning (GAL), which aims to find the most informative nodes in graphs for annotation to maximize the Graph Neural Networks (GNNs) performance, has attracted many research efforts but remains non-trivial challenges. One major challenge is that existing GAL strategies may introduce semantic confusion to the selected training set, particularly when graphs are noisy. Specifically, most existing methods assume all aggregating features to be helpful, ignoring the semantically negative effect between inter-class edges under the message-passing mechanism. In this work, we present Semantic-aware Active learning framework for Graphs (SAG) to mitigate the semantic confusion problem. Pairwise similarities and dissimilarities of nodes with semantic features are introduced to jointly evaluate the node influence. A new prototype-based criterion and query policy are also designed to maintain diversity and class balance of the selected nodes, respectively. Extensive experiments on the public benchmark graphs and a real-world financial dataset demonstrate that SAG significantly improves node classification performances and consistently outperforms previous methods. Moreover, comprehensive analysis and ablation study also verify the effectiveness of the proposed framework. Tianmeng Yang, Min Zhou 0006, Yujing Wang 0002, Zhengjie Lin, Lujia Pan, Bin Cui 0001, Yunhai Tong |
CIKM | 3 |
| 2023 | UPRISE: Universal Prompt Retrieval for Improving Zero-Shot EvaluationabstractDaixuan Cheng, Shaohan Huang, Junyu Bi, Yuefeng Zhan, Jianfeng Liu, Yujing Wang, Hao Sun, Furu Wei, Weiwei Deng, Qi Zhang. Proceedings of the 2023 Conference on Empirical Methods in Natural Language Processing. 2023. Daixuan Cheng, Shaohan Huang, Junyu Bi, Yuefeng Zhan, Yujing Wang 0002, Hao Sun 0015, Furu Wei, Qi Zhang 0066 |
EMNLP | 6 |
| 2023 | VL-Match: Enhancing Vision-Language Pretraining with Token-Level and Instance-Level MatchingabstractVision-Language Pretraining (VLP) has significantly improved the performance of various vision-language tasks with the matching of images and texts. In this paper, we propose VL-Match, a Vision-Language framework with Enhanced Token-level and Instance-level Matching. At the token level, a Vision-Language Replaced Token Detection task is designed to boost the substantial interaction between text tokens and images, where the text encoder of VLP works as a generator to generate a corrupted text, and the multimodal encoder of VLP works as a discriminator to predict whether each text token in the corrupted text matches the image. At the instance level, in the Image-Text Matching task that judges whether an image-text pair is matched, we propose a novel bootstrapping method to generate hard negative text samples that are different from the positive ones only at the token level. In this way, we can force the network to detect fine-grained differences between images and texts. Notably, with a smaller amount of parameters, VL-Match significantly outperforms previous SOTA on all image-text retrieval tasks. Junyu Bi, Daixuan Cheng, Ping Yao, Bochen Pang, Yuefeng Zhan, Chuanguang Yang, Yujing Wang 0002, Hao Sun 0015, Qi Zhang 0066 |
ICCV | 7 |
| 2023 | SpaceEvo: Hardware-Friendly Search Space Design for Efficient INT8 InferenceabstractThe combination of Neural Architecture Search (NAS) and quantization has proven successful in automatically designing low-FLOPs INT8 quantized neural networks (QNN). However, directly applying NAS to design accurate QNN models that achieve low latency on real-world devices leads to inferior performance. In this work, we identify that the poor INT8 latency is due to the quantization-unfriendly issue: the operator and configuration (e.g., channel width) choices in prior art search spaces lead to diverse quantization efficiency and can slow down the INT8 inference speed. To address this challenge, we propose SpaceEvo, an automatic method for designing a dedicated, quantization-friendly search space for each target hardware. The key idea of SpaceEvo is to automatically search hardware-preferred operators and configurations to construct the search space, guided by a metric called Q-T score to quantify how quantization-friendly a candidate search space is. We further train a quantized-for-all supernet over our discovered search space, enabling the searched models to be directly deployed without extra retraining or quantization. Our discovered models, SEQnet, establish new SOTA INT8 quantized accuracy under various latency constraints, achieving up to 10.1% accuracy improvement on ImageNet than prior art CNNs under the same latency. Extensive experiments on real devices show that SpaceEvo consistently outperforms manually-designed search spaces with up to 2.5× faster speed while achieving the same accuracy. Li Lyna Zhang, Jiahang Xu, Quanlu Zhang, Yujing Wang 0002, Yuqing Yang 0001, Ningxin Zheng, Ting Cao 0003, Mao Yang 0004 |
ICCV | 5 |
| 2023 | Constraint-aware and Ranking-distilled Token Pruning for Efficient Transformer InferenceabstractDeploying pre-trained transformer models like BERT on downstream tasks in resource-constrained scenarios is challenging due to their high inference cost, which grows rapidly with input sequence length. In this work, we propose a constraint-aware and ranking-distilled token pruning method ToP, which selectively removes unnecessary tokens as input sequence passes through layers, allowing the model to improve online inference speed while preserving accuracy. ToP overcomes the limitation of inaccurate token importance ranking in the conventional self-attention mechanism through a ranking-distilled token distillation technique, which distills effective token rankings from the final layer of unpruned models to early layers of pruned models. Then, ToP introduces a coarse-to-fine pruning approach that automatically selects the optimal subset of transformer layers and optimizes token pruning decisions within these layers through improved L0 regularization. Extensive experiments on GLUE benchmark and SQuAD tasks demonstrate that ToP outperforms state-of-the-art token pruning and model compression methods with improved accuracy and speedups. ToP reduces the average FLOPs of BERT by 8.1X while achieving competitive accuracy on GLUE, and provides a real latency speedup of up to 7.4X on an Intel CPU. Code is available at https://github.com/microsoft/Moonlit/tree/main/ToP Li Lyna Zhang, Jiahang Xu, Yujing Wang 0002, Shaoguang Yan, Yunqing Xia, Yuqing Yang 0001, Ting Cao 0003, Hao Sun 0015, Qi Zhang 0066, Mao Yang 0004 |
KDD | 4 |
| 2023 | Model-enhanced Vector IndexabstractEmbedding-based retrieval methods construct vector indices to search for document representations that are most similar to the query representations. They are widely used in document retrieval due to low latency and decent recall performance. Recent research indicates that deep retrieval solutions offer better model quality, but are hindered by unacceptable serving latency and the inability to support document updates. In this paper, we aim to enhance the vector index with end-to-end deep generative models, leveraging the differentiable advantages of deep retrieval models while maintaining desirable serving efficiency. We propose Model-enhanced Vector Index (MEVI), a differentiable model-enhanced index empowered by a twin-tower representation model. MEVI leverages a Residual Quantization (RQ) codebook to bridge the sequence-to-sequence deep retrieval and embedding-based models. To substantially reduce the inference time, instead of decoding the unique document ids in long sequential steps, we first generate some semantic virtual cluster ids of candidate documents in a small number of steps, and then leverage the well-adapted embedding vectors to further perform a fine-grained search for the relevant documents in the candidate virtual clusters. We empirically show that our model achieves better performance on the commonly used academic benchmarks MSMARCO Passage and Natural Questions, with comparable serving latency to dense retrieval solutions. Hailin Zhang 0004, Yujing Wang 0002, Qi Chen 0009, Ruiheng Chang, Ting Zhang 0002, Ziming Miao, Yingyan Hou, Xupeng Miao, Bochen Pang, Yuefeng Zhan, Hao Sun 0015, Qi Zhang 0066, Fan Yang 0024, Xing Xie 0001, Mao Yang 0004, Bin Cui 0001 |
NeurIPS | 2 |
| 2023 | Convolution-Enhanced Evolving Attention NetworksabstractAttention-based neural networks, such as Transformers, have become ubiquitous in numerous applications, including computer vision, natural language processing, and time-series analysis. In all kinds of attention networks, the attention maps are crucial as they encode semantic dependencies between input tokens. However, most existing attention networks perform modeling or reasoning based on representations, wherein the attention maps of different layers are learned separately without explicit interactions. In this paper, we propose a novel and generic evolving attention mechanism, which directly models the evolution of inter-token relationships through a chain of residual convolutional modules. The major motivations are twofold. On the one hand, the attention maps in different layers share transferable knowledge, thus adding a residual connection can facilitate the information flow of inter-token relationships across layers. On the other hand, there is naturally an evolutionary trend among attention maps at different abstraction levels, so it is beneficial to exploit a dedicated convolution-based module to capture this process. Equipped with the proposed mechanism, the convolution-enhanced evolving attention networks achieve superior performance in various applications, including time-series representation, natural language understanding, machine translation, and image classification. Especially on time-series representation tasks, Evolving Attention-enhanced Dilated Convolutional (EA-DC-) Transformer outperforms state-of-the-art models significantly, achieving an average of 17% improvement compared to the best SOTA. To the best of our knowledge, this is the first work that explicitly models the layer-wise evolution of attention maps. Our implementation is available at https://github.com/pkuyym/EvolvingAttention. Yujing Wang 0002, Yaming Yang 0001, Jiangang Bai, Mingliang Zhang 0004, Xiangtai Li, Jing Yu 0007, Ce Zhang 0001, Gao Huang 0001, Yunhai Tong |
IEEE Trans. Pattern Anal. Mach. Intell. | 1 |
| 2022 | SwiftPruner: Reinforced Evolutionary Pruning for Efficient Ad RelevanceabstractAd relevance modeling plays a critical role in online advertising systems including Microsoft Bing. To leverage powerful transformers like BERT in this low-latency setting, many existing approaches perform ad-side computations offline. While efficient, these approaches are unable to serve cold start ads, resulting in poor relevance predictions for such ads. This work aims to design a new, low-latency BERT via structured pruning to empower real-time online inference for cold start ads relevance on a CPU platform. Our challenge is that previous methods typically prune all layers of the transformer to a high, uniform sparsity, thereby producing models which cannot achieve satisfactory inference speed with an acceptable accuracy. Li Lyna Zhang, Youkow Homma, Yujing Wang 0002, Mao Yang 0004, Ruofei Zhang, Ting Cao 0003 |
CIKM | 3 |
| 2022 | Enhancing Self-Attention with Knowledge-Assisted Attention MapsabstractJiangang Bai, Yujing Wang, Hong Sun, Ruonan Wu, Tianmeng Yang, Pengfei Tang, Defu Cao, Mingliang Zhang1, Yunhai Tong, Yaming Yang, Jing Bai, Ruofei Zhang, Hao Sun, Wei Shen. Proceedings of the 2022 Conference of the North American Chapter of the Association for Computational Linguistics: Human Language Technologies. 2022. Jiangang Bai, Yujing Wang 0002, Ruonan Wu, Tianmeng Yang, Defu Cao, Mingliang Zhang 0004, Yunhai Tong, Yaming Yang 0001, Jing Bai 0010, Ruofei Zhang, Hao Sun 0015 |
NAACL-HLT | 2 |
| 2022 | A Neural Corpus Indexer for Document RetrievalabstractCurrent state-of-the-art document retrieval solutions mainly follow an index-retrieve paradigm, where the index is hard to be directly optimized for the final retrieval target. In this paper, we aim to show that an end-to-end deep neural network unifying training and indexing stages can significantly improve the recall performance of traditional methods. To this end, we propose Neural Corpus Indexer (NCI), a sequence-to-sequence network that generates relevant document identifiers directly for a designated query. To optimize the recall performance of NCI, we invent a prefix-aware weight-adaptive decoder architecture, and leverage tailored techniques including query generation, semantic document identifiers, and consistency-based regularization. Empirical studies demonstrated the superiority of NCI on two commonly used academic benchmarks, achieving +21.4% and +16.8% relative enhancement for Recall@1 on NQ320k dataset and R-Precision on TriviaQA dataset, respectively, compared to the best baseline method. Yujing Wang 0002, Yingyan Hou, Ziming Miao, Shibin Wu, Qi Chen 0009, Yuqing Xia, Chengmin Chi, Guoshuai Zhao 0001, Zheng Liu 0011, Xing Xie 0001, Hao Sun 0015, Qi Zhang 0066, Mao Yang 0004 |
NeurIPS | 1 |
| 2022 | Learning Multi-granularity Consecutive User Intent Unit for Session-based RecommendationabstractSession-based recommendation aims to predict a user's next action based on previous actions in the current session. The major challenge is to capture authentic and complete user preferences in the entire session. Recent work utilizes graph structure to represent the entire session and adopts Graph Neural Network (GNN) to encode session information. This modeling choice has been proved to be effective and achieved remarkable results. However, most of the existing studies only consider each item within the session independently and do not capture session semantics from a high-level perspective. Such limitation often leads to severe information loss and increases the difficulty of capturing long-range dependencies within a session. Intuitively, compared with individual items, a session snippet, i.e., a group of locally consecutive items, is able to provide supplemental user intents which are hardly captured by existing methods. In this work, we propose to learn multi-granularity consecutive user intent unit to improve the recommendation performance. Specifically, we creatively propose Multi-granularity Intent Heterogeneous Session Graph (MIHSG) which captures the interactions between different granularity intent units and relieves the burden of long-dependency. Moreover, we propose the Intent Fusion Ranking (IFR) module to compose the recommendation results from various granularity user intents. Compared with current methods that only leverage intents from individual items, IFR benefits from different granularity user intents to generate more accurate and comprehensive session representation, thus eventually boosting recommendation performance. We conduct extensive experiments on five session-based recommendation datasets and the results demonstrate the effectiveness of our method. Compared to current state-of-the-art methods, we achieve as large as 10.21% gain on [email protected] and 15.53% gain on [email protected] Jiayan Guo, Yaming Yang 0001, Xiangchen Song, Yuan Zhang 0024, Yujing Wang 0002, Jing Bai 0010, Yan Zhang 0004 |
WSDM | 5 |
| 2021 | Syntax-BERT: Improving Pre-trained Transformers with Syntax TreesabstractJiangang Bai, Yujing Wang, Yiren Chen, Yaming Yang, Jing Bai, Jing Yu, Yunhai Tong. Proceedings of the 16th Conference of the European Chapter of the Association for Computational Linguistics: Main Volume. 2021. Jiangang Bai, Yujing Wang 0002, Yaming Yang 0001, Jing Bai 0010, Jing Yu 0007, Yunhai Tong |
EACL | 2 |
| 2021 | Evolving Attention with Residual ConvolutionsabstractTransformer is a ubiquitous model for natural language processing and has attracted wide attentions in computer vision. The attention maps are indispensable for a transformer model to encode the dependencies among input tokens. However, they are learned independently in each layer and sometimes fail to capture precise patterns. In this paper, we propose a novel and generic mechanism based on evolving attention to improve the performance of transformers. On one hand, the attention maps in different layers share common knowledge, thus the ones in preceding layers can instruct the attention in succeeding layers through residual connections. On the other hand, low-level and high-level attentions vary in the level of abstraction, so we adopt convolutional layers to model the evolutionary process of attention maps. The proposed evolving attention mechanism achieves significant performance improvement over various state-of-the-art models for multiple tasks, including image classification, natural language understanding and machine translation. Yujing Wang 0002, Yaming Yang 0001, Jiangang Bai, Mingliang Zhang 0004, Jing Bai 0010, Jing Yu 0007, Ce Zhang 0001, Gao Huang 0001, Yunhai Tong |
ICML | 1 |
| 2021 | CogTree: Cognition Tree Loss for Unbiased Scene Graph GenerationabstractScene graphs are semantic abstraction of images that encourage visual understanding and reasoning. However, the performance of Scene Graph Generation (SGG) is unsatisfactory when faced with biased data in real-world scenarios. Conventional debiasing research mainly studies from the view of balancing data distribution or learning unbiased models and representations, ignoring the correlations among the biased classes. In this work, we analyze this problem from a novel cognition perspective: automatically building a hierarchical cognitive structure from the biased predictions and navigating that hierarchy to locate the relationships, making the tail relationships receive more attention in a coarse-to-fine mode. To this end, we propose a novel debiasing Cognition Tree (CogTree) loss for unbiased SGG. We first build a cognitive structure CogTree to organize the relationships based on the prediction of a biased SGG model. The CogTree distinguishes remarkably different relationships at first and then focuses on a small portion of easily confused ones. Then, we propose a debiasing loss specially for this cognitive structure, which supports coarse-to-fine distinction for the correct relationships. The loss is model-agnostic and consistently boosting the performance of several state-of-the-art models. The code is available at: https://github.com/CYVincent/Scene-Graph-Transformer-CogTree. Jing Yu 0007, Yujing Wang 0002, Yue Hu 0002, Qi Wu 0001 |
IJCAI | 3 |
| 2021 | DeGNN: Improving Graph Neural Networks with Graph DecompositionabstractMining from graph-structured data is an integral component of graph data management. A recent trending technique, graph convolutional network (GCN), has gained momentum in the graph mining field, and plays an essential part in numerous graph-related tasks. Although the emerging GCN optimization techniques bring improvements to specific scenarios, they perform diversely in different applications and introduce many trial-and-error costs for practitioners. Moreover, existing GCN models often suffer from oversmoothing problem. Besides, the entanglement of various graph patterns could lead to non-robustness and harm the final performance of GCNs. In this work, we propose a simple yet efficient graph decomposition approach to improve the performance of general graph neural networks. We first empirically study existing graph decomposition methods and propose an automatic connectivity-ware graph decomposition algorithm, DeGNN. To provide a theoretical explanation, we then characterize GCN from the information-theoretic perspective and show that under certain conditions, the mutual information between the output after l layers and the input of GCN converges to 0 exponentially with respect to l. On the other hand, we show that graph decomposition can potentially weaken the condition of such convergence rate, alleviating the information loss when GCN becomes deeper. Extensive experiments on various academic benchmarks and real-world production datasets demonstrate that graph decomposition generally boosts the performance of GNN models. Moreover, our proposed solution DeGNN achieves state-of-the-art performances on almost all these tasks. Xupeng Miao, Nezihe Merve Gürel, Wentao Zhang 0001, Zhichao Han 0001, Bo Li 0026, Wei Min, Susie Xi Rao, Hansheng Ren, Yinan Shan, Yingxia Shao, Fan Wu 0011, Hui Xue 0004, Yaming Yang 0001, Zitao Zhang, Shuai Zhang 0007, Yujing Wang 0002, Bin Cui 0001, Ce Zhang 0001 |
KDD | 18 |
| 2020 | TextNAS: A Neural Architecture Search Space Tailored for Text RepresentationabstractLearning text representation is crucial for text classification and other language related tasks. There are a diverse set of text representation networks in the literature, and how to find the optimal one is a non-trivial problem. Recently, the emerging Neural Architecture Search (NAS) techniques have demonstrated good potential to solve the problem. Nevertheless, most of the existing works of NAS focus on the search algorithms and pay little attention to the search space. In this paper, we argue that the search space is also an important human prior to the success of NAS in different applications. Thus, we propose a novel search space tailored for text representation. Through automatic search, the discovered network architecture outperforms state-of-the-art models on various public datasets on text classification and natural language inference tasks. Furthermore, some of the design principles found in the automatic network agree well with human intuition. Yujing Wang 0002, Yaming Yang 0001, Jing Bai 0010, Ce Zhang 0001, Guinan Su, Xiaoyu Kou, Yunhai Tong, Mao Yang 0004, Lidong Zhou |
AAAI | 1 |
| 2020 | AutoADR: Automatic Model Design for Ad RelevanceabstractLarge-scale pre-trained models have attracted extensive attention in the research community and shown promising results on various tasks of natural language processing. However, these pre-trained models are memory and computation intensive, hindering their deployment into industrial online systems like Ad Relevance. Meanwhile, how to design an effective yet efficient model architecture is another challenging problem in online Ad Relevance. Recently, AutoML shed new lights on architecture design, but how to integrate it with pre-trained language models remains unsettled. In this paper, we propose AutoADR (Automatic model design for AD Relevance) --- a novel end-to-end framework to address this challenge, and share our experience to ship these cutting-edge techniques into online Ad Relevance system at Microsoft Bing. Specifically, AutoADR leverages a one-shot neural architecture search algorithm to find a tailored network architecture for Ad Relevance. The search process is simultaneously guided by knowledge distillation from a large pre-trained teacher model (e.g. BERT), while taking the online serving constraints (e.g. memory and latency) into consideration. We add the model designed by AutoADR as a sub-model into the production Ad Relevance model. This additional sub-model improves the Precision-Recall AUC (PR AUC) on top of the original Ad Relevance model by 2.65X of the normalized shipping bar. More importantly, adding this automatically designed sub-model leads to a statistically significant 4.6% Bad-Ad ratio reduction in online A/B testing. This model has been shipped into Microsoft Bing Ad Relevance Production model. Yaming Yang 0001, Yujing Wang 0002, Yunhai Tong, Jing Bai 0010, Ruofei Zhang |
CIKM | 4 |
| 2020 | LadaBERT: Lightweight Adaptation of BERT through Hybrid Model CompressionabstractBERT is a cutting-edge language representation model pre-trained by a large corpus, which achieves superior performances on various natural language understanding tasks. However, a major blocking issue of applying BERT to online services is that it is memory-intensive and leads to unsatisfactory latency of user requests, raising the necessity of model compression. Existing solutions leverage the knowledge distillation framework to learn a smaller model that imitates the behaviors of BERT. However, the training procedure of knowledge distillation is expensive itself as it requires sufficient training data to imitate the teacher model. In this paper, we address this issue by proposing a tailored solution named LadaBERT (Lightweight adaptation of BERT through hybrid model compression), which combines the advantages of different model compression methods, including weight pruning, matrix factorization and knowledge distillation. LadaBERT achieves state-of-the-art accuracy on various public datasets while the training overheads can be reduced by an order of magnitude. Yihuan Mao, Yujing Wang 0002, Chufan Wu, Chen Zhang 0001, Yang Wang 0053, Quanlu Zhang, Yaming Yang 0001, Yunhai Tong, Jing Bai 0010 |
COLING | 2 |
| 2020 | Multivariate Time-series Anomaly Detection via Graph Attention NetworkabstractAnomaly detection on multivariate time-series is of great importance in both data mining research and industrial applications. Recent approaches have achieved significant progress in this topic, but there is remaining limitations. One major limitation is that they do not capture the relationships between different time-series explicitly, resulting in inevitable false alarms. In this paper, we propose a novel self-supervised framework for multivariate time-series anomaly detection to address this issue. Our framework considers each univariate time-series as an individual feature and includes two graph attention layers in parallel to learn the complex dependencies of multivariate time-series in both temporal and feature dimensions. In addition, our approach jointly optimizes a forecasting-based model and a reconstruction-based model, obtaining better time-series representations through a combination of single-timestamp prediction and reconstruction of the entire time-series. We demonstrate the efficacy of our model through extensive experiments. The proposed method outperforms other state-of-the-art models on three real-world datasets. Further analysis shows that our method has good interpretability and is useful for anomaly diagnosis. Yujing Wang 0002, Juanyong Duan, Congrui Huang, Defu Cao, Yunhai Tong, Bixiong Xu, Jing Bai 0010, Jie Tong, Qi Zhang 0066 |
ICDM | 2 |
| 2020 | Mucko: Multi-Layer Cross-Modal Knowledge Reasoning for Fact-based Visual Question AnsweringabstractFact-based Visual Question Answering (FVQA) requires external knowledge beyond the visible content to answer questions about an image. This ability is challenging but indispensable to achieve general VQA. One limitation of existing FVQA solutions is that they jointly embed all kinds of information without fine-grained selection, which introduces unexpected noises for reasoning the final answer. How to capture the question-oriented and information-complementary evidence remains a key challenge to solve the problem. In this paper, we depict an image by a multi-modal heterogeneous graph, which contains multiple layers of information corresponding to the visual, semantic and factual features. On top of the multi-layer graph representations, we propose a modality-aware heterogeneous graph convolutional network to capture evidence from different layers that is most relevant to the given question. Specifically, the intra-modal graph convolution selects evidence from each modality and cross-modal graph convolution aggregates relevant information across different graph layers. By stacking this process multiple times, our model performs iterative reasoning across three modalities and predicts the optimal answer by analyzing all question-oriented evidence. We achieve a new state-of-the-art performance on the FVQA task and demonstrate the effectiveness and interpretability of our model with extensive experiments. Jing Yu 0007, Yujing Wang 0002, Yajing Sun, Yue Hu 0002, Qi Wu 0001 |
IJCAI | 3 |
| 2020 | Spectral Temporal Graph Neural Network for Multivariate Time-series ForecastingabstractMultivariate time-series forecasting plays a crucial role in many real-world applications. It is a challenging problem as one needs to consider both intra-series temporal correlations and inter-series correlations simultaneously. Recently, there have been multiple works trying to capture both correlations, but most, if not all of them only capture temporal correlations in the time domain and resort to pre-defined priors as inter-series relationships. In this paper, we propose Spectral Temporal Graph Neural Network (StemGNN) to further improve the accuracy of multivariate time-series forecasting. StemGNN captures inter-series correlations and temporal dependencies jointly in the spectral domain. It combines Graph Fourier Transform (GFT) which models inter-series correlations and Discrete Fourier Transform (DFT) which models temporal dependencies in an end-to-end framework. After passing through GFT and DFT, the spectral representations hold clear patterns and can be predicted effectively by convolution and sequential learning modules. Moreover, StemGNN learns inter-series correlations automatically from the data without using pre-defined priors. We conduct extensive experiments on ten real-world datasets to demonstrate the effectiveness of StemGNN. Defu Cao, Yujing Wang 0002, Juanyong Duan, Ce Zhang 0001, Congrui Huang, Yunhai Tong, Bixiong Xu, Jing Bai 0010, Jie Tong, Qi Zhang 0066 |
NeurIPS | 2 |
| 2020 | Cross-modal knowledge reasoning for knowledge-based visual question answering
Jing Yu 0007, Yujing Wang 0002, Weifeng Zhang 0002, Yue Hu 0002, Jianlong Tan |
Pattern Recognit. | 3 |
| 2019 | Time-Series Anomaly Detection Service at MicrosoftabstractLarge companies need to monitor various metrics (for example, Page Views and Revenue) of their applications and services in real time. At Microsoft, we develop a time-series anomaly detection service which helps customers to monitor the time-series continuously and alert for potential incidents on time. In this paper, we introduce the pipeline and algorithm of our anomaly detection service, which is designed to be accurate, efficient and general. The pipeline consists of three major modules, including data ingestion, experimentation platform and online compute. To tackle the problem of time-series anomaly detection, we propose a novel algorithm based on Spectral Residual (SR) and Convolutional Neural Network (CNN). Our work is the first attempt to borrow the SR model from visual saliency detection domain to time-series anomaly detection. Moreover, we innovatively combine SR and CNN together to improve the performance of SR model. Our approach achieves superior experimental results compared with state-of-the-art baselines on both public datasets and Microsoft production data. Hansheng Ren, Bixiong Xu, Yujing Wang 0002, Chao Yi, Congrui Huang, Xiaoyu Kou, Tony Xing, Mao Yang 0004, Jie Tong, Qi Zhang 0066 |
KDD | 3 |
| 2013 | Ranking Scientific Articles by Exploiting Citations, Authors, Journals, and Time InformationabstractRanking scientific articles is an important but challenging task, partly due to the dynamic nature of the evolving publication network. In this paper, we mainly focus on two problems: (1) how to rank articles in the heterogeneous network; and (2) how to use time information in the dynamic network in order to obtain a better ranking result. To tackle the problems, we propose a graph based ranking method, which utilizes citations, authors, journals/conferences and the publication time information collaboratively. The experiments were carried out on two public datasets. The result shows that our approach is practical and ranks scientific articles more accurately than the state-of-art methods. Yujing Wang 0002, Yunhai Tong, Ming Zeng 0009 |
AAAI | 1 |