EDBT 2026 Demo / reviewers in the wild / expert
Taifeng Wang
dblp:01/1483
· DBLP profile ↗
40ranked-venue papers
1as first author
13since 2021 · last 2026
0009-0007-1116-0228ORCID · corroborated
Domains — the database's venue-derived domains; a paper can count in several
Artificial intelligence and machine learning · 34 · 1 first-author · 13 since 2021Databases, data management, data science and information retrieval · 10 · 1 first-author · 2 since 2021Graphics, computer vision, multimedia, augmented reality and games · 7
| Year | Publication | Venue | Position |
|---|---|---|---|
| 2026 | TiKMiX: Efficient Semi-Dynamic Data Mixture via Data Influence for LLM Pre-trainingabstractYifan Wang, Binbinliu, Fengze Liu, Yuanfan Guo, Jiyao Deng, Xuecheng Wu, Weidong Zhou, Xiaohuan Zhou, Taifeng Wang. Proceedings of the 64th Annual Meeting of the Association for Computational Linguistics (Volume 1: Long Papers). 2026. Fengze Liu, Yuanfan Guo, Jiyao Deng, Xiaohuan Zhou, Taifeng Wang |
ACL (1) | 9 |
| 2025 | Size-Generalizable RNA Structure Evaluation by Exploring Hierarchical GeometriesabstractUnderstanding the 3D structure of RNA is essential for deciphering its function and developing RNA-based therapeutics. Geometric Graph Neural Networks (GeoGNNs) that conform to the $\mathrm{E}(3)$-symmetry have advanced RNA structure evaluation, a crucial step toward RNA structure prediction. However, existing GeoGNNs are still defective in two aspects: 1. inefficient or incapable of capturing the full geometries of RNA; 2. limited generalization ability when the size of RNA significantly differs between training and test datasets. In this paper, we propose EquiRNA, a novel equivariant GNN model by exploring the three-level hierarchical geometries of RNA. At its core, EquiRNA effectively addresses the size generalization challenge by reusing the representation of nucleotide, the common building block shared across RNAs of varying sizes. Moreover, by adopting a scalarization-based equivariant GNN as the backbone, our model maintains directional information while offering higher computational efficiency compared to existing GeoGNNs. Additionally, we propose a size-insensitive $K$-nearest neighbor sampling strategy to enhance the model's robustness to RNA size shifts. We test our approach on our created benchmark as well as an existing dataset. The results show that our method significantly outperforms other state-of-the-art methods, providing a robust baseline for RNA 3D structure modeling and evaluation. Zongzhao Li, Jiacheng Cen, Wenbing Huang 0001, Taifeng Wang |
ICLR | 4 |
| 2025 | MuRating: A High Quality Data Selecting Approach to Multilingual Large Language Model PretrainingabstractData quality is a critical driver of large language model performance, yet existing model-based selection methods focus almost exclusively on English, neglecting other languages that are essential in the training mix for multilingual LLMs. We introduce MuRating, a scalable framework that transfers high-quality English data-quality signals into a multilingual autorater, capable of handling 17 languages. MuRating aggregates multiple English autoraters via pairwise comparisons to learn unified document quality scores, then projects these judgments through translation to train a multilingual evaluator on monolingual, cross-lingual, and parallel text pairs. Applied to web data, MuRating selects balanced subsets of English and multilingual content to pretrain LLaMA-architecture models of 1.2B and 7B parameters. Compared to strong baselines, including QuRater, FineWeb2-HQ, AskLLM, DCLM, our approach increases average accuracy on both English benchmarks and multilingual evaluations. Extensive analyses further validate that pairwise training provides greater stability and robustness than pointwise scoring, underscoring the effectiveness of MuRating as a general multilingual data-selection framework. Zhixun Chen, Wenhan Han, Binbin Li 0001, Haobin Lin, Fengze Liu, Bingni Zhang, Taifeng Wang, Trevor Cohn |
NeurIPS | 10 |
| 2025 | Exploring Polyglot Harmony: On Multilingual Data Allocation for Large Language Models PretrainingabstractLarge language models (LLMs) have become integral to a wide range of applications worldwide, driving an unprecedented global demand for effective multilingual capabilities. Central to achieving robust multilingual performance is the strategic allocation of language proportions within training corpora. However, determining optimal language ratios is highly challenging due to intricate cross-lingual interactions and sensitivity to dataset scale. This paper introduces CLIMB (Cross-Lingual Interaction-aware Multilingual Balancing), a novel framework designed to systematically optimize multilingual data allocation. At its core, CLIMB introduces a cross-lingual interaction-aware language ratio, explicitly quantifying each language’s effective allocation by capturing inter-language dependencies. Leveraging this ratio, CLIMB proposes a principled two-step optimization procedure—first equalizing marginal benefits across languages, then maximizing the magnitude of the resulting language allocation vectors—significantly simplifying the inherently complex multilingual optimization problem. Extensive experiments confirm that CLIMB can accurately measure cross-lingual interactions across various multilingual settings. LLMs trained with CLIMB-derived proportions consistently achieve state-of-the-art multilingual performance, even achieve competitive performance with open-sourced LLMs trained with more tokens. Yubing Ren, Fengze Liu, Haobin Lin, Bingni Zhang, Taifeng Wang |
NeurIPS | 8 |
| 2025 | MoORE: SVD-based Model MoE-ization for Conflict- and Oblivion-Resistant Multi-Task AdaptationabstractAdapting large-scale foundation models in multi-task scenarios often suffers from task conflict and oblivion.
To mitigate such issues, we propose a novel "model MoE-ization" strategy that leads to a conflict- and oblivion-resistant multi-task adaptation method.
Given a weight matrix of a pre-trained model, our method applies SVD to it and introduces a learnable router to adjust its singular values based on tasks and samples.
Accordingly, the weight matrix becomes a Mixture of Orthogonal Rank-one Experts (MoORE), in which each expert corresponds to the outer product of a left singular vector and the corresponding right one.
We can improve the model capacity by imposing a learnable orthogonal transform on the right singular vectors.
Unlike low-rank adaptation (LoRA) and its MoE-driven variants, MoORE guarantees the experts' orthogonality and maintains the column space of the original weight matrix.
These two properties make the adapted model resistant to the conflicts among the new tasks and the oblivion of its original tasks, respectively.
Experiments on various datasets demonstrate that MoORE outperforms existing multi-task adaptation methods consistently, showing its superiority in terms of conflict- and oblivion-resistance.
The code is available at https://github.com/DaShenZi721/MoORE. Shen Yuan, Taifeng Wang, Hongteng Xu |
NeurIPS | 3 |
| 2024 | LogicMP: A Neuro-symbolic Approach for Encoding First-order Logic ConstraintsabstractIntegrating first-order logic constraints (FOLCs) with neural networks is a crucial but challenging problem since it involves modeling intricate correlations to satisfy the constraints. This paper proposes a novel neural layer, LogicMP, which performs mean-field variational inference over a Markov Logic Network (MLN). It can be plugged into any off-the-shelf neural network to encode FOLCs while retaining modularity and efficiency. By exploiting the structure and symmetries in MLNs, we theoretically demonstrate that our well-designed, efficient mean-field iterations greatly mitigate the difficulty of MLN inference, reducing the inference from sequential calculation to a series of parallel tensor operations. Empirical results in three kinds of tasks over images, graphs, and text show that LogicMP outperforms advanced competitors in both performance and efficiency. Weidi Xu, Lele Xie, Jianshan He, Hongting Zhou, Taifeng Wang, Xiaopei Wan, Jingdong Chen, Chao Qu |
ICLR | 6 |
| 2023 | xTrimoGene: An Efficient and Scalable Representation Learner for Single-Cell RNA-Seq DataabstractAdvances in high-throughput sequencing technology have led to significant progress in measuring gene expressions at the single-cell level. The amount of publicly available single-cell RNA-seq (scRNA-seq) data is already surpassing 50M records for humans with each record measuring 20,000 genes. This highlights the need for unsupervised representation learning to fully ingest these data, yet classical transformer architectures are prohibitive to train on such data in terms of both computation and memory. To address this challenge, we propose a novel asymmetric encoder-decoder transformer for scRNA-seq data, called xTrimoGene$^\alpha$ (or xTrimoGene for short), which leverages the sparse characteristic of the data to scale up the pre-training. This scalable design of xTrimoGene reduces FLOPs by one to two orders of magnitude compared to classical transformers while maintaining high accuracy, enabling us to train the largest transformer models over the largest scRNA-seq dataset today. Our experiments also show that the performance of xTrimoGene improves as we scale up the model sizes, and it also leads to SOTA performance over various downstream tasks, such as cell type annotation, perturb-seq effect prediction, and drug combination prediction.
xTrimoGene model is now available for use as a service via the following link: https://api.biomap.com/xTrimoGene/apply. Minsheng Hao, Xingyi Cheng, Chiming Liu, Jianzhu Ma, Xuegong Zhang, Taifeng Wang |
NeurIPS | 8 |
| 2022 | Keywords and Instances: A Hierarchical Contrastive Learning Framework Unifying Hybrid Granularities for Text GenerationabstractMingzhe Li, XieXiong Lin, Xiuying Chen, Jinxiong Chang, Qishen Zhang, Feng Wang, Taifeng Wang, Zhongyi Liu, Wei Chu, Dongyan Zhao, Rui Yan. Proceedings of the 60th Annual Meeting of the Association for Computational Linguistics (Volume 1: Long Papers). 2022. Mingzhe Li 0001, Xiexiong Lin, Xiuying Chen, Jinxiong Chang, Qishen Zhang, Feng Wang 0023, Taifeng Wang, Zhongyi Liu 0001, Dongyan Zhao 0001, Rui Yan 0001 |
ACL (1) | 7 |
| 2022 | A Logic Aware Neural Generation Method for Explainable Data-to-textabstractThe most notable neural data-to-text approaches generate natural language from structural data relying on the surface form of the structural content, which ignores the underlying logical correlation between the input data and the target text. Moreover, identifying such logical associations and explaining them in natural language is desirable but not yet studied. In this paper, we introduce a practical data-to-text method for the logic-critical scenario, specifically for anti-money laundering applications. It involves detecting risks from input data and explaining any abnormal behaviors in natural language. The proposed method is a Logic Aware Neural Generation framework (LANG), which is a preliminary attempt to explore the integration of logic modeling and text generation. Concretely, we first convert expert rules to a logic graph. Then, the model utilizes meta path based encoder to exploit the expert knowledge. Besides, a retriever module with the encoded logic knowledge is used to bridge the gap between numeric input and target text. Finally, a rule-constrained loss is leveraged to improve the generation probability of tokens in rule recalled statements to ensure accuracy. We conduct extensive experiments on anti-money laundering data. Results show that the proposed method significantly outperforms baselines in both objective measures with relative 35% improvements in F1 score and subjective measures with 30% improvement in human preference. Xiexiong Lin, Huaisong Li, Linlin Chao, Fuzhen Zhuang, Taifeng Wang |
KDD | 7 |
| 2021 | PairRE: Knowledge Graph Embeddings via Paired Relation VectorsabstractLinlin Chao, Jianshan He, Taifeng Wang, Wei Chu. Proceedings of the 59th Annual Meeting of the Association for Computational Linguistics and the 11th International Joint Conference on Natural Language Processing (Volume 1: Long Papers). 2021. Linlin Chao, Jianshan He, Taifeng Wang |
ACL/IJCNLP (1) | 3 |
| 2021 | Document-level Event Extraction via Parallel Prediction NetworksabstractHang Yang, Dianbo Sui, Yubo Chen, Kang Liu, Jun Zhao, Taifeng Wang. Proceedings of the 59th Annual Meeting of the Association for Computational Linguistics and the 11th International Joint Conference on Natural Language Processing (Volume 1: Long Papers). 2021. Dianbo Sui, Yubo Chen 0001, Kang Liu 0001, Jun Zhao 0001, Taifeng Wang |
ACL/IJCNLP (1) | 6 |
| 2021 | Multi-Sentence Argument Linking via An Event-Aware Hierarchical EncoderabstractMulti-sentence argument linking aims at detecting implicit event arguments across sentences, which is indispensable when textual events span across multiple sentences in a document. Previous studies suffer from the inherent limitations of error propagation and lack the explicit modeling of the local and non-local interactions in a textual event. In this paper, we propose an event-aware hierarchical encoder for multi-sentence argument linking. Specifically, we introduce a hierarchical encoder to explicitly capture the local and global interactions in a textual event. Furthermore, we introduce an auxiliary task to predict the event-relevant context in a manner of multi-task learning, which can implicitly benefit the argument linking model to be aware of the event-relevant context. The empirical results on the widely used argument linking dataset show that our model significantly outperforms the baselines, which demonstrates the effectiveness of our proposed method. Yubo Chen 0001, Kang Liu 0001, Jun Zhao 0001, Taifeng Wang |
CIKM | 5 |
| 2021 | Probing into the Root: A Dataset for Reason Extraction of Structural Events from Financial DocumentsabstractThis paper proposes a new task regarding event reason extraction from document-level texts.Unlike the previous causality detection task, we do not assign target events in the text but only provide structural event descriptions, and such settings accord more with practice scenarios.Moreover, we annotate a large dataset FinReason for evaluation, which provides Reasons annotation for Financial events in company announcements.This task is challenging because the cases of multiple-events, multiple-reasons, and implicit-reasons are included.In total, FinReason contains 8,794 documents, 12,861 financial events and 11,006 reason spans.We also provide the performance of existing canonical methods in event extraction and machine reading comprehension on this task.The results show a 7 percentage point F1 score gap between the best model and human performance, and existing methods are far from resolving this problem. Kang Liu 0001, Yubo Chen 0001, Taifeng Wang, Jun Zhao 0001 |
EACL | 4 |
| 2020 | SpellGCN: Incorporating Phonological and Visual Similarities into Language Models for Chinese Spelling CheckabstractChinese Spelling Check (CSC) is a task to detect and correct spelling errors in Chinese natural language.Existing methods have made attempts to incorporate the similarity knowledge between Chinese characters.However, they take the similarity knowledge as either an external input resource or just heuristic rules.This paper proposes to incorporate phonological and visual similarity knowledge into language models for CSC via a specialized graph convolutional network (SpellGCN).The model builds a graph over the characters, and SpellGCN is learned to map this graph into a set of inter-dependent character classifiers.These classifiers are applied to the representations extracted by another network, such as BERT, enabling the whole network to be end-to-end trainable.Experiments 1 are conducted on three human-annotated datasets.Our method achieves superior performance against previous models by a large margin. Xingyi Cheng, Weidi Xu, Kunlong Chen, Shaohua Jiang, Taifeng Wang, Yuan Qi 0001 |
ACL | 6 |
| 2020 | Generating Informative Conversational Response using Recurrent Knowledge-Interaction and Knowledge-CopyabstractKnowledge-driven conversation approaches have achieved remarkable research attention recently. However, generating an informative response with multiple relevant knowledge without losing fluency and coherence is still one of the main challenges. To address this issue, this paper proposes a method that uses recurrent knowledge interaction among response decoding steps to incorporate appropriate knowledge. Furthermore, we introduce a knowledge copy mechanism using a knowledge-aware pointer network to copy words from external knowledge according to knowledge attention distribution. Our joint neural conversation model which integrates recurrent Knowledge-Interaction and knowledge Copy (KIC) performs well on generating informative responses. Experiments demonstrate that our model with fewer parameters yields significant improvements over competitive baselines on two datasets Wizard-of-Wikipedia(average Bleu +87%; abs.: 0.034) and DuConv(average Bleu +20%; abs.: 0.047)) with different knowledge formats (textual & structured) and different languages (English & Chinese). Xiexiong Lin, Weiyu Jian, Jianshan He, Taifeng Wang |
ACL | 4 |
| 2020 | Towards Fast and Accurate Neural Chinese Word Segmentation with Multi-Criteria LearningabstractThe ambiguous annotation criteria lead to divergence of Chinese Word Segmentation (CWS) datasets in various granularities.Multi-criteria Chinese word segmentation aims to capture various annotation criteria among datasets and leverage their common underlying knowledge.In this paper, we propose a domain adaptive segmenter to exploit diverse criteria of various datasets.Our model is based on Bidirectional Encoder Representations from Transformers (BERT), which is responsible for introducing open-domain knowledge.Private and shared projection layers are proposed to capture domain-specific knowledge and common knowledge, respectively.We also optimize computational efficiency via distillation, quantization, and compiler optimization.Experiments show that our segmenter outperforms the previous state of the art (SOTA) models on 10 CWS datasets with superior efficiency. Weipeng Huang, Xingyi Cheng, Kunlong Chen, Taifeng Wang |
COLING | 4 |
| 2020 | Incremental Event Detection via Knowledge Consolidation NetworksabstractConventional approaches to event detection usually require a fixed set of pre-defined event types. Such a requirement is often challenged in real-world applications, as new events continually occur. Due to huge computation cost and storage budge, it is infeasible to store all previous data and re-train the model with all previous data and new data, every time new events arrive. We formulate such challenging scenarios as incremental event detection, which requires a model to learn new classes incrementally without performance degradation on previous classes. However, existing incremental learning methods cannot handle semantic ambiguity and training data imbalance problems between old and new classes in the task of incremental event detection. In this paper, we propose a Knowledge Consolidation Network (KCN) to address the above issues. Specifically, we devise two components, prototype enhanced retrospection and hierarchical distillation, to mitigate the adverse effects of semantic ambiguity and class imbalance, respectively. Experimental results demonstrate the effectiveness of the proposed method, outperforming the state-of-the-art model by 19% and 13.4% of whole F1 score on ACE benchmark and TAC KBP benchmark, respectively. Yubo Chen 0001, Jun Zhao 0001, Taifeng Wang |
EMNLP (1) | 4 |
| 2020 | Question Directed Graph Attention Network for Numerical Reasoning over TextabstractKunlong Chen, Weidi Xu, Xingyi Cheng, Zou Xiaochuan, Yuyu Zhang, Le Song, Taifeng Wang, Yuan Qi, Wei Chu. Proceedings of the 2020 Conference on Empirical Methods in Natural Language Processing (EMNLP). 2020. Kunlong Chen, Weidi Xu, Xingyi Cheng, Zou Xiaochuan, Yuyu Zhang, Taifeng Wang, Yuan Qi 0001 |
EMNLP (1) | 7 |
| 2020 | Symmetric Regularization based BERT for Pair-wise Semantic ReasoningabstractThe ability of semantic reasoning over the sentence pair is essential for many natural language understanding tasks, e.g., natural language inference and machine reading comprehension. A recent significant improvement in these tasks comes from BERT. As reported, the next sentence prediction (NSP) in BERT is of great significance for downstream problems with sentence-pair input. Despite its effectiveness, NSP still lacks the essential signal to distinguish between entailment and shallow correlation. To remedy this, we propose to augment the NSP task to a multi-class categorization task, which includes previous sentence prediction (PSP). This task encourages the model to learn the subtle semantics, thereby improves the ability of semantic understanding. Furthermore, by using a smoothing technique, the scopes of NSP and PSP are expanded into a broader range which includes close but nonsuccessive sentences. This simple method yields remarkable improvement against vanilla BERT. Our method consistently improves the performance on the NLI and MRC benchmarks by a large margin, including the challenging HANS dataset. Weidi Xu, Xingyi Cheng, Kunlong Chen, Taifeng Wang |
SIGIR | 4 |
| 2019 | Variational Semi-Supervised Aspect-Term Sentiment Analysis via TransformerabstractAspect-term sentiment analysis (ATSA) is a long-standing challenge in natural language processing.It requires fine-grained semantical reasoning about a target entity appeared in the text.As manual annotation over the aspects is laborious and time-consuming, the amount of labeled data is limited for supervised learning.This paper proposes a semisupervised method for the ATSA problem by using the Variational Autoencoder based on Transformer.The model learns the latent distribution via variational inference.By disentangling the latent representation into the aspect-specific sentiment and the lexical context, our method induces the underlying sentiment prediction for the unlabeled data, which then benefits the ATSA classifier.Our method is classifier-agnostic, i.e., the classifier is an independent module and various supervised models can be integrated.Experimental results are obtained on the SemEval 2014 task 4 and show that our method is effective with different five specific classifiers and outperforms these models by a significant margin. Xingyi Cheng, Weidi Xu, Taifeng Wang, Weipeng Huang, Kunlong Chen |
CoNLL | 3 |
| 2019 | BERT-Based Multi-head Selection for Joint Entity-Relation Extraction
Weipeng Huang, Xingyi Cheng, Taifeng Wang |
NLPCC (2) | 3 |
| 2019 | Large-Scale Low-Rank Matrix Learning with Nonconvex RegularizersabstractLow-rank modeling has many important applications in computer vision and machine learning. While the matrix rank is often approximated by the convex nuclear norm, the use of nonconvex low-rank regularizers has demonstrated better empirical performance. However, the resulting optimization problem is much more challenging. Recent state-of-the-art requires an expensive full SVD in each iteration. In this paper, we show that for many commonly-used nonconvex low-rank regularizers, the singular values obtained from the proximal operator can be automatically threshold. This allows the proximal operator to be efficiently approximated by the power method. We then develop a fast proximal algorithm and its accelerated variant with inexact proximal step. It can be guaranteed that the squared distance between consecutive iterates converges at a rate of $O(1/T)$O(1/T), where $T$T is the number of iterations. Furthermore, we show the proposed algorithm can be parallelized, and the resultant algorithm achieves nearly linear speedup w.r.t. the number of threads. Extensive experiments are performed on matrix completion and robust principal component analysis. Significant speedup over the state-of-the-art is observed. Quanming Yao, James T. Kwok, Taifeng Wang, Tie-Yan Liu |
IEEE Trans. Pattern Anal. Mach. Intell. | 3 |
| 2017 | Asynchronous Stochastic Proximal Optimization Algorithms with Variance ReductionabstractRegularized empirical risk minimization (R-ERM) is an important branch of machine learning, since it constrains the capacity of the hypothesis space and guarantees the generalization ability of the learning algorithm. Two classic proximal optimization algorithms, i.e., proximal stochastic gradient descent (ProxSGD) and proximal stochastic coordinate descent (ProxSCD) have been widely used to solve the R-ERM problem. Recently, variance reduction technique was proposed to improve ProxSGD and ProxSCD, and the corresponding ProxSVRG and ProxSVRCD have better convergence rate. These proximal algorithms with variance reduction technique have also achieved great success in applications at small and moderate scales. However, in order to solve large-scale R-ERM problems and make more practical impacts, the parallel versions of these algorithms are sorely needed. In this paper, we propose asynchronous ProxSVRG (Async-ProxSVRG) and asynchronous ProxSVRCD (Async-ProxSVRCD) algorithms, and prove that Async-ProxSVRG can achieve near linear speedup when the training data is sparse, while Async-ProxSVRCD can achieve near linear speedup regardless of the sparse condition, as long as the number of block partitions are appropriately set. We have conducted experiments on a regularized logistic regression task. The results verified our theoretical findings and demonstrated the practical efficiency of the asynchronous stochastic proximal algorithms with variance reduction. Wei Chen 0034, Jingcheng Yu, Taifeng Wang, Zhiming Ma, Tie-Yan Liu |
AAAI | 4 |
| 2017 | Generalization Error Bounds for Optimization Algorithms via StabilityabstractMany machine learning tasks can be formulated as Regularized Empirical Risk Minimization (R-ERM), and solved by optimization algorithms such as gradient descent (GD), stochastic gradient descent (SGD), and stochastic variance reduction (SVRG). Conventional analysis on these optimization algorithms focuses on their convergence rates during the training process, however, people in the machine learning community may care more about the generalization performance of the learned model on unseen test data. In this paper, we investigate on this issue, by using stability as a tool. In particular, we decompose the generalization error for R-ERM, and derive its upper bound for both convex and nonconvex cases. In convex cases, we prove that the generalization error can be bounded by the convergence rate of the optimization algorithm and the stability of the R-ERM process, both in expectation (in the order of Yue Wang 0017, Wei Chen 0034, Taifeng Wang, Zhiming Ma, Tie-Yan Liu |
AAAI | 4 |
| 2017 | Asynchronous Stochastic Gradient Descent with Delay CompensationabstractWith the fast development of deep learning, it has become common to learn big neural networks using massive training data. Asynchronous Stochastic Gradient Descent (ASGD) is widely adopted to fulfill this task for its efficiency, which is, however, known to suffer from the problem of delayed gradients. That is, when a local worker adds its gradient to the global model, the global model may have been updated by other workers and this gradient becomes “delayed”. We propose a novel technology to compensate this delay, so as to make the optimization behavior of ASGD closer to that of sequential SGD. This is achieved by leveraging Taylor expansion of the gradient function and efficient approximators to the Hessian matrix of the loss function. We call the new algorithm Delay Compensated ASGD (DC-ASGD). We evaluated the proposed algorithm on CIFAR-10 and ImageNet datasets, and the experimental results demonstrate that DC-ASGD outperforms both synchronous SGD and asynchronous SGD, and nearly approaches the performance of sequential SGD. Shuxin Zheng, Taifeng Wang, Wei Chen 0034, Nenghai Yu, Zhiming Ma, Tie-Yan Liu |
ICML | 3 |
| 2017 | LightGBM: A Highly Efficient Gradient Boosting Decision TreeabstractGradient Boosting Decision Tree (GBDT) is a popular machine learning algorithm, and has quite a few effective implementations such as XGBoost and pGBRT. Although many engineering optimizations have been adopted in these implementations, the efficiency and scalability are still unsatisfactory when the feature dimension is high and data size is large. A major reason is that for each feature, they need to scan all the data instances to estimate the information gain of all possible split points, which is very time consuming. To tackle this problem, we propose two novel techniques: \emph{Gradient-based One-Side Sampling} (GOSS) and \emph{Exclusive Feature Bundling} (EFB). With GOSS, we exclude a significant proportion of data instances with small gradients, and only use the rest to estimate the information gain. We prove that, since the data instances with larger gradients play a more important role in the computation of information gain, GOSS can obtain quite accurate estimation of the information gain with a much smaller data size. With EFB, we bundle mutually exclusive features (i.e., they rarely take nonzero values simultaneously), to reduce the number of features. We prove that finding the optimal bundling of exclusive features is NP-hard, but a greedy algorithm can achieve quite good approximation ratio (and thus can effectively reduce the number of features without hurting the accuracy of split point determination by much). We call our new GBDT implementation with GOSS and EFB \emph{LightGBM}. Our experiments on multiple public datasets show that, LightGBM speeds up the training process of conventional GBDT by up to over 20 times while achieving almost the same accuracy. Guolin Ke, Thomas Finley, Taifeng Wang, Wei Chen 0034, Weidong Ma, Qiwei Ye, Tie-Yan Liu |
NIPS | 4 |
| 2016 | Asynchronous Accelerated Stochastic Gradient Descent
Wei Chen 0034, Jingcheng Yu, Taifeng Wang, Zhiming Ma, Tie-Yan Liu |
IJCAI | 4 |
| 2016 | A Communication-Efficient Parallel Algorithm for Decision TreeabstractDecision tree (and its extensions such as Gradient Boosting Decision Trees and Random Forest) is a widely used machine learning algorithm, due to its practical effectiveness and model interpretability. With the emergence of big data, there is an increasing need to parallelize the training process of decision tree. However, most existing attempts along this line suffer from high communication costs. In this paper, we propose a new algorithm, called \emph{Parallel Voting Decision Tree (PV-Tree)}, to tackle this challenge. After partitioning the training data onto a number of (e.g., $M$) machines, this algorithm performs both local voting and global voting in each iteration. For local voting, the top-$k$ attributes are selected from each machine according to its local data. Then, the indices of these top attributes are aggregated by a server, and the globally top-$2k$ attributes are determined by a majority voting among these local candidates. Finally, the full-grained histograms of the globally top-$2k$ attributes are collected from local machines in order to identify the best (most informative) attribute and its split point. PV-Tree can achieve a very low communication cost (independent of the total number of attributes) and thus can scale out very well. Furthermore, theoretical analysis shows that this algorithm can learn a near optimal decision tree, since it can find the best attribute with a large probability. Our experiments on real-world datasets show that PV-Tree significantly outperforms the existing parallel decision tree algorithms in the tradeoff between accuracy and efficiency. Guolin Ke, Taifeng Wang, Wei Chen 0034, Qiwei Ye, Zhiming Ma, Tie-Yan Liu |
NIPS | 3 |
| 2016 | Ada-Sal Network: emulate the Human Visual System
Yunong Wang, Nenghai Yu, Taifeng Wang |
Signal Process. Image Commun. | 3 |
| 2016 | A Ranking Approach on Large-Scale Graph With Multidimensional Heterogeneous InformationabstractGraph-based ranking has been extensively studied and frequently applied in many applications, such as webpage ranking. It aims at mining potentially valuable information from the raw graph-structured data. Recently, with the proliferation of rich heterogeneous information (e.g., node/edge features and prior knowledge) available in many real-world graphs, how to effectively and efficiently leverage all information to improve the ranking performance becomes a new challenging problem. Previous methods only utilize part of such information and attempt to rank graph nodes according to link-based methods, of which the ranking performances are severely affected by several well-known issues, e.g., over-fitting or high computational complexity, especially when the scale of graph is very large. In this paper, we address the large-scale graph-based ranking problem and focus on how to effectively exploit rich heterogeneous information of the graph to improve the ranking performance. Specifically, we propose an innovative and effective semi-supervised PageRank (SSP) approach to parameterize the derived information within a unified semi-supervised learning framework (SSLF-GR), then simultaneously optimize the parameters and the ranking scores of graph nodes. Experiments on the real-world large-scale graphs demonstrate that our method significantly outperforms the algorithms that consider such graph information only partially. Wei Wei 0002, Bin Gao 0001, Tie-Yan Liu, Taifeng Wang, Guohui Li 0001, Hang Li 0001 |
IEEE Trans. Cybern. | 4 |
| 2015 | Improve Neural Network Using Saliency
Yunong Wang, Nenghai Yu, Taifeng Wang |
ICIG (2) | 3 |
| 2014 | Sequential Click Prediction for Sponsored Search with Recurrent Neural NetworksabstractClick prediction is one of the fundamental problems in sponsored search. Most of existing studies took advantage of machine learning approaches to predict ad click for each event of ad view independently. However, as observed in the real-world sponsored search system, user's behaviors on ads yield high dependency on how the user behaved along with the past time, especially in terms of what queries she submitted, what ads she clicked or ignored, and how long she spent on the landing pages of clicked ads, etc. Inspired by these observations, we introduce a novel framework based on Recurrent Neural Networks (RNN). Compared to traditional methods, this framework directly models the dependency on user's sequential behaviors into the click prediction process through the recurrent structure in RNN. Large scale evaluations on the click-through logs from a commercial search engine demonstrate that our approach can significantly improve the click prediction accuracy, compared to sequence-independent approaches. Yuyu Zhang, Hanjun Dai, Chang Xu 0008, Taifeng Wang, Jiang Bian 0002, Bin Wang 0004, Tie-Yan Liu |
AAAI | 5 |
| 2014 | Sampling dilemma: towards effective data sampling for click prediction in sponsored searchabstractPrecise prediction of the probability that users click on ads plays a key role in sponsored search. State-of-the-art sponsored search systems typically employ a machine learning approach to conduct click prediction. While paying much attention to extracting useful features and building effective models, previous studies have overshadowed seemingly less obvious but essentially important challenges in terms of data sampling. To fulfill the learning objective of click prediction, it is not only necessary to ensure that the sampled training data implies the similar input distribution compared with the real world one, but also to guarantee that the sampled training data yield the consistent conditional output distribution, i.e. click-through rate (CTR), with the real world data. However, due to the sparseness of clicks in sponsored search, it is a bit contradictory to address these two challenges simultaneously. In this paper, we first take a theoretical analysis to reveal this sampling dilemma, followed by a thorough data analysis which demonstrates that the straightforward random sampling method may not be effective to balance these two kinds of consistency in sampling dilemma simultaneously. To address this problem, we propose a new sampling algorithm which can succeed in retaining the consistency between the sampled data and real world in terms of both input distribution and conditional output distribution. Large scale evaluations on the click-through logs from a commercial search engine demonstrate that this new sampling algorithm can effectively address the sampling dilemma. Further experiments illustrate that, by using the training data obtained by our new sampling algorithm, we can learn the model with much higher accuracy in click prediction. Jiang Bian 0002, Taifeng Wang, Wei Chen 0034, Xiaoyan Zhu 0001, Tie-Yan Liu |
WSDM | 3 |
| 2013 | Psychological advertising: exploring user psychology for click prediction in sponsored searchabstractPrecise click prediction is one of the key components in the sponsored search system. Previous studies usually took advantage of two major kinds of information for click prediction, i.e., relevance information representing the similarity between ads and queries and historical click-through information representing users' previous preferences on the ads. These existing works mainly focused on interpreting ad clicks in terms of what users seek (i.e., relevance information) and how users choose to click (historically clicked-through information). However, few of them attempted to understand why users click the ads. In this paper, we aim at answering this ``why'' question. In our opinion, users click those ads that can convince them to take further actions, and the critical factor is if those ads can trigger users' desires in their hearts. Our data analysis on a commercial search engine reveals that specific text patterns, e.g., ``official site'', ``$x\%$ off'', and ``guaranteed return in $x$ days'', are very effective in triggering users' desires, and therefore lead to significant differences in terms of click-through rate (CTR). These observations motivate us to systematically model user psychological desire in order for a precise prediction on ad clicks. To this end, we propose modeling user psychological desire in sponsored search according to Maslow's desire theory, which categorizes psychological desire into five levels and each one is represented by a set of textual patterns automatically mined from ad texts. We then construct novel features for both ads and users based on our definition on psychological desire and incorporate them into the learning framework of click prediction. Large scale evaluations on the click-through logs from a commercial search engine demonstrate that this approach can result in significant improvement in terms of click prediction accuracy, for both the ads with rich historical data and those with rare one. Further analysis reveals that specific pattern combinations are especially effective in driving click-through rates, which provides a good guideline for advertisers to improve their ad textual descriptions. Taifeng Wang, Jiang Bian 0002, Yuyu Zhang, Tie-Yan Liu |
KDD | 1 |
| 2012 | Large-scale graph mining and learning for information retrievalabstractFor many information retrieval applications, we need to deal with the ranking problem on very large scale graphs. However, it is non-trivial to perform efficient and effective ranking on them. On one aspect, we need to design scalable algorithms. On another aspect, we also need to develop powerful computational infrastructure to support these algorithms. This tutorial aims at giving a timely introduction to the promising advances in the aforementioned aspects in recent years, and providing the audiences with a comprehensive view on the related literature. Bin Gao 0001, Taifeng Wang, Tie-Yan Liu |
SIGIR | 2 |
| 2012 | Relational click prediction for sponsored searchabstractThis paper is concerned with the prediction of clicking an ad in sponsored search. The accurate prediction of user's click on an ad plays an important role in sponsored search, because it is widely used in both ranking and pricing of the ads. Previous work on click prediction usually takes a single ad as input, and ignores its relationship to the other ads shown in the same page. This independence assumption here, however, might not be valid in the real scenario. In this paper, we first perform an analysis on this issue by looking at the click-through rates (CTR) of the same ad, in the same position and for the same query, but surrounded by different ads. We found that in most cases the CTR varies largely, which suggests that the relationship between ads is really an important factor in predicting click probability. Furthermore, our investigation shows that the more similar the surrounding ads are to an ad, the lower the CTR of the ad is. Based on this observation, we design a continuous conditional random fields (CRF) based model for click prediction, which considers both the features of an ad and its similarity to the surrounding ads. We show that the model can be effectively learned using maximum likelihood estimation, and can also be efficiently inferred due to its closed form solution. Our experimental results on the click-through log from a commercial search engine show that the proposed model can predict clicks more accurately than previous independent models. To our best knowledge this is the first work that predicts ad clicks by considering the relationship between ads. Chenyan Xiong, Taifeng Wang, Wenkui Ding, Yidong Shen, Tie-Yan Liu |
WSDM | 2 |
| 2011 | Semi-supervised ranking on very large graphs with rich metadataabstractGraph ranking plays an important role in many applications, such as page ranking on web graphs and entity ranking on social networks. In applications, besides graph structure, rich information on nodes and edges and explicit or implicit human supervision are often available. In contrast, conventional algorithms (e.g., PageRank and HITS) compute ranking scores by only resorting to graph structure information. A natural question arises here, that is, how to effectively and efficiently leverage all the information to more accurately calculate graph ranking scores than the conventional algorithms, assuming that the graph is also very large. Previous work only partially tackled the problem, and the proposed solutions are also not satisfying. This paper addresses the problem and proposes a general framework as well as an efficient algorithm for graph ranking. Specifically, we define a semi-supervised learning framework for ranking of nodes on a very large graph and derive within our proposed framework an efficient algorithm called Semi-Supervised PageRank. In the algorithm, the objective function is defined based upon a Markov random walk on the graph. The transition probability and the reset probability of the Markov model are defined as parametric models based on features on nodes and edges. By minimizing the objective function, subject to a number of constraints derived from supervision information, we simultaneously learn the optimal parameters of the model and the optimal ranking scores of the nodes. Finally, we show that it is possible to make the algorithm efficient to handle a billion-node graph by taking advantage of the sparsity of the graph and implement it in the MapReduce logic. Experiments on real data from a commercial search engine show that the proposed algorithm can outperform previous algorithms on several tasks. Bin Gao 0001, Tie-Yan Liu, Wei Wei 0002, Taifeng Wang, Hang Li 0001 |
KDD | 4 |
| 2011 | Page importance computation based on Markov processes
Bin Gao 0001, Tie-Yan Liu, Yuting Liu 0002, Taifeng Wang, Zhiming Ma, Hang Li 0001 |
Inf. Retr. | 4 |
| 2009 | A general markov framework for page importance computationabstractWe propose a General Markov Framework for computing page importance. Under the framework, a Markov Skeleton Process is used to model the random walk conducted by the web surfer on a given graph. Page importance is then defined as the product of page reachability and page utility, which can be computed from the transition probability and the mean staying time of the pages in the Markov Skeleton Process respectively. We show that this general framework can cover many existing algorithms as its special cases, and that the framework can help us define new algorithms to handle more complex problems. In particular, we demonstrate the use of the framework with the exploitation of a new process named Mirror Semi-Markov Process. The experimental results validate that the Mirror Semi-Markov Process model is more effective than previous models in several tasks. Bin Gao 0001, Tie-Yan Liu, Zhiming Ma, Taifeng Wang, Hang Li 0001 |
CIKM | 4 |
| 2007 | A Search-Based Web Image Annotation MethodabstractAutomatic image annotation is an effective way for managing and retrieving abundant images on the internet. In this paper, we propose a novel search-based method for web image annotation. Firstly, surrounding text and other textual information in the hosting web pages are used as the candidate annotations. As the candidates are very noisy, two measures are defined to re-rank them and only top-ranked ones are reserved as the final annotations. One measure is based on the visual consistence between the given image and the image search results when the candidate annotation is submitted as a query. The other measure is based on the textual consistence between one candidate and others, which is calculated from the pairwise co-occurrence using image search engines. Finally, Dempster-Shafer multiple evidence combination approach is used to combine the visual and textual measures for the final ranking. Experimental results on web images demonstrate the effectiveness of the proposed method. Xiaoguang Rui, Nenghai Yu, Taifeng Wang, Mingjing Li |
ICME | 3 |