Demo Data from SIGIR 2026 / reviewers in the wild / expert
Chih-Jen Lin
dblp:61/1218
· DBLP profile ↗
120ranked-venue papers
14as first author
15since 2021 · last 2025
0000-0003-4684-8747ORCID · verified
Domains — the database's venue-derived domains; a paper can count in several
Artificial intelligence and machine learning · 88 · 11 first-author · 13 since 2021Databases, data management, data science and information retrieval · 41 · 2 first-author · 8 since 2021Systems, architecture and hardware · 7 · 2 first-authorGraphics, computer vision, multimedia, augmented reality and games · 6 · 1 since 2021Applied, interdisciplinary, general and emerging computing · 4Human-computer interaction and ubiquitous computing · 1
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
29 papers |
Optimization for machine learning · 27% Learning theory · 25% Trustworthy machine learning · 8% | |
| Databases, data mining, and information retrieval
12 papers |
Data mining · 38% Recommender systems · 34% Machine learning and data management · 18% | |
| Theoretical computer science
18 papers |
Mathematical optimization · 90% Algorithms and data structures · 8% Coding theory · 1% | |
| Computer architecture, parallel and distributed computing, and storage systems
7 papers |
Parallel and multicore computing · 28% High-performance computing · 27% Memory systems · 16% |
Topics — the 30 heaviest of 104, each with the papers that count most for it
| Topic | Weight | Papers | Last | Evidence papers |
|---|---|---|---|---|
Mathematical optimization › continuous optimization
convex optimization |
1.1 | 8 | 2019 | The Common-directions Method for Regularized Empirical Risk Minimization · J. Mach. Learn. Res. 2019 Iteration complexity of feasible descent methods for convex optimization · J. Mach. Learn. Res. 2014 An Improved GLMNET for L1-regularized Logistic Regression · J. Mach. Learn. Res. 2012 |
Machine learning › Learning theory › classification
multiclass classification |
1.0 | 4 | 2025 | Revisiting One-Versus-One and One-Versus-Rest: Insights into Imbalanced Multi-Class Classification · ICDM 2025 Generalized Bradley-Terry Models and Multi-Class Probability Estimates · J. Mach. Learn. Res. 2006 Probability Estimates for Multi-class Classification by Pairwise Coupling · J. Mach. Learn. Res. 2004 |
Data mining › predictive modeling
classification |
0.9 | 3 | 2024 | Exploring Space Efficiency in a Tree-based Linear Model for Extreme Multi-label Classification · EMNLP 2024 Experiences and lessons in developing industry-strength machine learning and data mining software · KDD 2012 Recent Advances of Large-Scale Linear Classification · Proc. IEEE 2012 |
Machine learning › Learning theory
classification |
0.9 | 2 | 2025 | Revisiting One-Versus-One and One-Versus-Rest: Insights into Imbalanced Multi-Class Classification · ICDM 2025 Probability Estimates for Multi-class Classification by Pairwise Coupling · J. Mach. Learn. Res. 2004 |
Machine learning › Learning paradigms
class imbalance |
0.9 | 1 | 2025 | Revisiting One-Versus-One and One-Versus-Rest: Insights into Imbalanced Multi-Class Classification · ICDM 2025 |
Machine learning › Deep learning architectures and training
loss function design |
0.9 | 1 | 2025 | Revisiting One-Versus-One and One-Versus-Rest: Insights into Imbalanced Multi-Class Classification · ICDM 2025 |
Machine learning › Efficient and distributed learning
sampled softmax |
0.9 | 1 | 2025 | Sampled Estimators For Softmax Must Be Biased · NeurIPS 2025 |
Machine learning › Learning theory › classification
linear classification |
0.8 | 3 | 2021 | Limited-memory Common-directions Method With Subsampled Newton Directions for Large-scale Linear Classification · ICDM 2021 Incremental and decremental training for linear classification · KDD 2014 Recent Advances of Large-Scale Linear Classification · Proc. IEEE 2012 |
Data mining › predictive modeling › classification › multi-label classification
extreme multi-label classification |
0.8 | 1 | 2024 | Exploring Space Efficiency in a Tree-based Linear Model for Extreme Multi-label Classification · EMNLP 2024 |
Machine learning › Optimization for machine learning
large-scale optimization |
0.7 | 3 | 2021 | Limited-memory Common-directions Method With Subsampled Newton Directions for Large-scale Linear Classification · ICDM 2021 Coordinate Descent Method for Large-scale L2-loss Linear Support Vector Machines · J. Mach. Learn. Res. 2008 A sequential dual method for large scale multi-class linear svms · KDD 2008 |
Machine learning › Graph learning
graph representation learning |
0.6 | 1 | 2022 | On the Use of Unrealistic Predictions in Hundreds of Papers Evaluating Graph Representations · AAAI 2022 |
Recommender systems › recommender system evaluation
off-policy evaluation |
0.6 | 1 | 2022 | Practical Counterfactual Policy Learning for Top-K Recommendations · KDD 2022 |
Recommender systems › collaborative filtering
matrix factorization |
0.5 | 2 | 2017 | A Unified Algorithm for One-Cass Structured Matrix Factorization with Side Information · AAAI 2017 LIBMF: A Library for Parallel Matrix Factorization in Shared-memory Systems · J. Mach. Learn. Res. 2016 |
Machine learning › Representation and self-supervised learning › representation learning
metric learning |
0.5 | 1 | 2021 | Efficient Optimization Methods for Extreme Similarity Learning with Nonlinear Embeddings · KDD 2021 |
Machine learning › Optimization for machine learning
second-order optimization |
0.5 | 1 | 2021 | Limited-memory Common-directions Method With Subsampled Newton Directions for Large-scale Linear Classification · ICDM 2021 |
Machine learning › Optimization for machine learning › second-order optimization
subsampled newton methods |
0.5 | 1 | 2021 | Limited-memory Common-directions Method With Subsampled Newton Directions for Large-scale Linear Classification · ICDM 2021 |
Machine learning › Optimization for machine learning › large-scale optimization
large-scale linear classification |
0.4 | 2 | 2016 | Parallel Dual Coordinate Descent Method for Large-scale Linear Classification in Multi-core Environments · KDD 2016 Recent Advances of Large-Scale Linear Classification · Proc. IEEE 2012 |
Machine learning › Optimization for machine learning
regularized risk minimization |
0.4 | 1 | 2019 | The Common-directions Method for Regularized Empirical Risk Minimization · J. Mach. Learn. Res. 2019 |
Mathematical optimization
continuous optimization |
0.4 | 3 | 2021 | Limited-memory Common-directions Method With Subsampled Newton Directions for Large-scale Linear Classification · ICDM 2021 Recent Advances of Large-Scale Linear Classification · Proc. IEEE 2012 Coordinate Descent Method for Large-scale L2-loss Linear Support Vector Machines · J. Mach. Learn. Res. 2008 |
Machine learning › Optimization for machine learning › coordinate descent
dual coordinate ascent |
0.3 | 2 | 2016 | Parallel Dual Coordinate Descent Method for Large-scale Linear Classification in Multi-core Environments · KDD 2016 A sequential dual method for large scale multi-class linear svms · KDD 2008 |
Machine learning › Probabilistic and Bayesian machine learning › statistical inference › regression › generalized linear model
logistic regression |
0.3 | 2 | 2015 | Fast Matrix-Vector Multiplications for Large-Scale Logistic Regression on Shared-Memory Systems · ICDM 2015 Trust region Newton methods for large-scale logistic regression · ICML 2007 |
Recommender systems
side information integration |
0.3 | 1 | 2017 | A Unified Algorithm for One-Cass Structured Matrix Factorization with Side Information · AAAI 2017 |
Machine learning › Kernel, tree and ensemble methods
support vector machine |
0.3 | 3 | 2010 | Training and Testing Low-degree Polynomial Data Mappings via Linear SVM · J. Mach. Learn. Res. 2010 Coordinate Descent Method for Large-scale L2-loss Linear Support Vector Machines · J. Mach. Learn. Res. 2008 A dual coordinate descent method for large-scale linear SVM · ICML 2008 |
Mathematical optimization › regularization › regularized optimization
l1-regularized logistic regression |
0.3 | 2 | 2012 | An Improved GLMNET for L1-regularized Logistic Regression · J. Mach. Learn. Res. 2012 An improved GLMNET for l1-regularized logistic regression · KDD 2011 |
Algorithms and data structures
classification |
0.3 | 1 | 2025 | Revisiting One-Versus-One and One-Versus-Rest: Insights into Imbalanced Multi-Class Classification · ICDM 2025 |
Recommender systems › collaborative filtering › matrix factorization
parallel matrix factorization |
0.2 | 1 | 2016 | LIBMF: A Library for Parallel Matrix Factorization in Shared-memory Systems · J. Mach. Learn. Res. 2016 |
Parallel and multicore computing › parallel computing
parallel machine learning |
0.2 | 1 | 2016 | Parallel Dual Coordinate Descent Method for Large-scale Linear Classification in Multi-core Environments · KDD 2016 |
Memory systems
shared memory |
0.2 | 1 | 2016 | LIBMF: A Library for Parallel Matrix Factorization in Shared-memory Systems · J. Mach. Learn. Res. 2016 |
Machine learning › Optimization for machine learning
coordinate descent |
0.2 | 2 | 2011 | An improved GLMNET for l1-regularized logistic regression · KDD 2011 Iterative Scaling and Coordinate Descent Methods for Maximum Entropy Models · J. Mach. Learn. Res. 2010 |
Machine learning and data management › scalable machine learning
out-of-core training |
0.2 | 2 | 2011 | Large Linear Classification When Data Cannot Fit in Memory · IJCAI 2011 Large linear classification when data cannot fit in memory · KDD 2010 |
Methods — techniques the papers use, named apart from their topics
coordinate descent · 1.9common-directions method · 1.8neural network · 1.7kernel SVM · 1.7accelerated gradient · 1.0BFGS · 1.0sampled-softmax · 0.9f-divergence · 0.9weight pruning · 0.8tree-based linear model · 0.8hyperparameter search · 0.7importance weighting · 0.6empirical analysis · 0.6counterfactual learning · 0.6subsampled newton directions · 0.5stochastic optimization · 0.5shared-memory parallelism · 0.5parallel matrix factorization · 0.5
| Year | Publication | Venue | Position |
|---|---|---|---|
| 2025 | SparseKmeans: Efficient K-means Clustering For Sparse DataabstractWe introduce SparseKmeans, the first Python package for fast K-means clustering on high-dimensional sparse data. Most existing K-means implementations, such as scikit-learn, are only optimized for dense data and do not run efficiently on sparse inputs. In this work, we thoroughly investigate how to accelerate widely used K-means algorithms on sparse data via matrix operations. In particular, we propose a new design of Elkan's method that aggregates distance computations and reduces fragmented memory access. By analyzing the structure of key matrices and leveraging highly optimized sparse matrix libraries, SparseKmeans achieves up to 9x speedup over scikit-learn. The package is available at https://github.com/cjlin1/sparsekmeans. Khoi Nguyen Pham Dang, He-Zhe Lin, Chih-Jen Lin |
CIKM | 3 |
| 2025 | Revisiting One-Versus-One and One-Versus-Rest: Insights into Imbalanced Multi-Class ClassificationabstractOne-versus-one (OVO) and one-versus-rest (OVR) are two widely adopted methods to decompose multi-class problems into several binary classification problems. It is well known that, in the case of kernel SVM, the two methods yield similar test accuracy. Thus, people generally assume that they differ mainly in training time and model size. However, our research reveals that if one considers an evaluation metric taking class imbalance into account, these two methods may give notable performance differences. To explore this phenomenon, we first conduct a detailed analysis of kernel SVM and then extend our study to neural networks. Additionally, we propose novel loss functions for neural networks that effectively integrate the OVO and OVR perspectives. Our experiments clearly demonstrate the robustness of OVO in handling imbalanced multi-class classification, highlighting its advantages over OVR in these challenging scenarios. Kuan-Ting Chen, Chih-Jen Lin |
ICDM | 2 |
| 2025 | Sampled Estimators For Softmax Must Be BiasedabstractModels requiring probabilistic outputs are ubiquitous and used in fields such as natural language processing, contrastive learning, and recommendation systems. The standard method of designing such a model is to output unconstrained logits, which are normalized into probabilities with the softmax function. The normalization involves computing a summation across all classes, which becomes prohibitively expensive for problems with a large number of classes. An important strategy to reduce the cost is to sum over a sampled subset of classes in the softmax function, known as the sampled softmax. It was known that the sampled softmax is biased; the expectation taken over the sampled classes is not equal to the softmax function. Many works focused on reducing the bias by using a better way of sampling the subset. However, while sampled softmax is biased, it is unclear whether an unbiased function different from sampled softmax exists. In this paper, we show that all functions that only access a sampled subset of classes must be biased.
With this result, we prevent efforts in finding unbiased loss functions and validate that past efforts devoted to reducing bias are the best we can do. Li-Chung Lin, Yaxu Liu, Chih-Jen Lin |
NeurIPS | 3 |
| 2025 | One-Class SVM Probabilistic OutputsabstractOne-class support vector machine (SVM) is an extension of SVM to handle unlabeled data. As a mature technique for outlier detection, one-class SVM has been widely used in many applications. However, similar to standard two-class SVM, the design of one-class SVM does not give probabilistic outputs. For two-class SVM, some methods have been proposed to effectively obtain probabilistic outputs, but due to the difficulty of no-label information, less attention has been paid to one-class SVM. Our aim in this work is to propose some practically viable techniques to generate probabilistic outputs for one-class SVM. We investigate existing methods for two-class SVM and explain why they may not be suitable for one-class SVM. Due to the lack of label information, we think a feasible setting is to have probabilities mimic to the decision values of training data. Based on this principle, we propose several new methods. Detailed experiments on both artificial and real-world data demonstrate the effectiveness of the proposed methods. Zhongyi Que, Chih-Jen Lin |
IEEE Trans. Neural Networks Learn. Syst. | 2 |
| 2024 | Exploring Space Efficiency in a Tree-based Linear Model for Extreme Multi-label ClassificationabstractExtreme multi-label classification (XMC) aims to identify relevant subsets from numerous labels.Among the various approaches for XMC, tree-based linear models are effective due to their superior efficiency and simplicity.However, the space complexity of tree-based methods is not well-studied.Many past works assume that storing the model is not affordable and apply techniques such as pruning to save space, which may lead to performance loss.In this work, we conduct both theoretical and empirical analyses on the space to store a tree model under the assumption of sparse data, a condition frequently met in text data.We found that some features may be unused when training binary classifiers in a tree method, resulting in zero values in the weight vectors.Hence, storing only non-zero elements can greatly save space.Our experimental results indicate that tree models can require less than 10% of the size of the standard one-vs-rest method for multi-label text classification.Our research provides a simple procedure to estimate the size of a tree model before training any classifier in the tree nodes.Then, if the model size is already acceptable, this approach can help avoid modifying the model through weight pruning or other techniques. He-Zhe Lin, Cheng-Hung Liu 0001, Chih-Jen Lin |
EMNLP | 3 |
| 2024 | One-class Matrix Factorization: Point-Wise Regression-Based or Pair-Wise Ranking-Based?abstractOne-class matrix factorization (MF) is an important technique for recommender systems with implicit feedback. In one widely used setting, a regression function is fit in a point-wise manner on observed and some unobserved (user, item) entries. Recently, in AAAI 2019, Chen et al. [2] proposed a pair-wise ranking-based approach for observed (user, item) entries to be compared against unobserved ones. They concluded that the pair-wise setting performs consistently better than the more traditional point-wise setting. However, after some detailed investigation, we explain by mathematical derivations that their method may perform only similar to the point-wise ones. We also identified some problems when reproducing their experimental results. After considering suitable settings, we rigorously compare point-wise and pair-wise one-class MFs, and show that the pair-wise method is actually not better. Therefore, for one-class MF, the more traditional and mature point-wise setting should still be considered. Our findings contradict the conclusions in [2] and serve as a call for caution when researchers are comparing between two machine learning methods. Sheng-Wei Chen 0003, Chih-Jen Lin |
RecSys | 2 |
| 2023 | On the Thresholding Strategy for Infrequent Labels in Multi-label ClassificationabstractIn multi-label classification, the imbalance between labels is often a concern. For a label that seldom occurs, the default threshold used to generate binarized predictions of that label is usually sub-optimal. However, directly tuning the threshold to optimize F-measure has been observed to overfit easily. In this work, we explain why this overfitting occurs. Then, we analyze the FBR heuristic, a previous technique proposed to address the overfitting issue. We explain its success but also point out some problems unobserved before. Then, we first propose a variant of the FBR heuristic that not only fixes the problems but is also more justifiable. Second, we propose a new technique based on smoothing the F-measure when tuning the threshold. We theoretically prove that, with proper parameters, smoothing results in desirable properties of the tuned threshold. Based on the idea of smoothing, we then propose jointly optimizing micro-F and macro-F as a lightweight alternative free from extra hyperparameters. Our methods are empirically evaluated on text and node classification datasets. The results show that our methods consistently outperform the FBR heuristic. Yu-Jen Lin, Chih-Jen Lin |
CIKM | 2 |
| 2023 | On the "Rough Use" of Machine Learning TechniquesabstractMachine learning is everywhere, but unfortunately, we are not experts of every method. Sometimes we "inappropriately'' use machine learning techniques. Examples include reporting training instead of test performance and comparing two methods without suitable hyper-parameter searches. However, the reality is that there are more sophisticated or more subtle examples, which we broadly call the "rough use'' of machine learning techniques. The setting may be roughly fine, but seriously speaking, is inappropriate. We briefly discuss two intriguing examples. Chih-Jen Lin |
SIGIR | 1 |
| 2022 | On the Use of Unrealistic Predictions in Hundreds of Papers Evaluating Graph RepresentationsabstractPrediction using the ground truth sounds like an oxymoron in machine learning. However, such an unrealistic setting was used in hundreds, if not thousands of papers in the area of finding graph representations. To evaluate the multi-label problem of node classification by using the obtained representations, many works assume that the number of labels of each test instance is known in the prediction stage. In practice such ground truth information is rarely available, but we point out that such an inappropriate setting is now ubiquitous in this research area. We detailedly investigate why the situation occurs. Our analysis indicates that with unrealistic information, the performance is likely over-estimated. To see why suitable predictions were not used, we identify difficulties in applying some multi-label techniques. For the use in future studies, we propose simple and effective settings without using practically unknown information. Finally, we take this chance to compare major graph representation learning methods on multi-label node classification. Li-Chung Lin, Cheng-Hung Liu 0001, Chih-Ming Chen 0003, Kai-Chin Hsu, I-Feng Wu, Ming-Feng Tsai, Chih-Jen Lin |
AAAI | 7 |
| 2022 | Practical Counterfactual Policy Learning for Top-K RecommendationsabstractFor building recommender systems, a critical task is to learn a policy with collected feedback (e.g., ratings, clicks) to decide which items to be recommended to users. However, it has been shown that the selection bias in the collected feedback leads to biased learning and thus a sub-optimal policy. To deal with this issue, counterfactual learning has received much attention, where existing approaches can be categorized as either value learning or policy learning approaches. This work studies policy learning approaches for top-K recommendations with a large item space and points out several difficulties related to importance weight explosion, observation insufficiency, and training efficiency. A practical framework for policy learning is then proposed to overcome these difficulties. Our experiments confirm the effectiveness and efficiency of the proposed framework. Yaxu Liu, Jui-Nan Yen, Bo-Wen Yuan, Rundong Shi, Chih-Jen Lin |
KDD | 6 |
| 2022 | Even the Simplest Baseline Needs Careful Re-investigation: A Case Study on XML-CNNabstractSi-An Chen, Jie-jyun Liu, Tsung-Han Yang, Hsuan-Tien Lin, Chih-Jen Lin. Proceedings of the 2022 Conference of the North American Chapter of the Association for Computational Linguistics: Human Language Technologies. 2022. Si-An Chen, Jie-Jyun Liu, Tsung-Han Yang, Hsuan-Tien Lin, Chih-Jen Lin |
NAACL-HLT | 5 |
| 2022 | A Study on Truncated Newton Methods for Linear ClassificationabstractTruncated Newton (TN) methods have been a useful technique for large-scale optimization. Instead of obtaining the full Newton direction, a truncated method approximately solves the Newton equation with an inner conjugate gradient (CG) procedure (TNCG for the whole method). These methods have been employed to efficiently solve linear classification problems. However, even in this deeply studied field, various theoretical and numerical aspects were not completely explored. The first contribution of this work is to comprehensively study the global and local convergence when TNCG is applied to linear classification. Because of the lack of twice differentiability under some losses, many past works cannot be applied here. We prove various missing pieces of theory from scratch and clarify many proper references. The second contribution is to study the termination of the CG method. For the first time when TNCG is applied to linear classification, we show that the inner stopping condition strongly affects the convergence speed. We propose using a quadratic stopping criterion to achieve both robustness and efficiency. The third contribution is that of combining the study on inner stopping criteria with that of preconditioning. We discuss how convergence theory is affected by preconditioning and finally propose an effective preconditioned TNCG. Leonardo Galli, Chih-Jen Lin |
IEEE Trans. Neural Networks Learn. Syst. | 2 |
| 2021 | Limited-memory Common-directions Method With Subsampled Newton Directions for Large-scale Linear ClassificationabstractThe common-directions method is an optimization method recently proposed to utilize second-order information. It is especially efficient on large-scale linear classification problems, and it is competitive with state-of-the-art optimization methods like BFGS, LBFGS, and Nesterov’s accelerated gradient method. The main idea of the method is to minimize the local quadratic approximation within the selected subspace. Regarding the selection of the subspace, the original authors only focused on the span of current and past gradient directions. In this work, we analyze the impact of subspace selection, and point out that the lack of direction diversity can be a potential weakness for using gradients as directions. To address this problem, we propose the use of subsampled Newton directions, which always possess diversity unless they are already close to the true Newton direction. Our experiments on large-scale linear classification problems show that our proposed methods are generally better than subsampled Newton methods and the original common-directions method. Jui-Nan Yen, Chih-Jen Lin |
ICDM | 2 |
| 2021 | Efficient Optimization Methods for Extreme Similarity Learning with Nonlinear EmbeddingsabstractWe study the problem of learning similarity by using nonlinear embedding models (e.g., neural networks) from all possible pairs. This problem is well-known for its difficulty of training with the extreme number of pairs. For the special case of using linear embeddings, many studies have addressed this issue of handling all pairs by considering certain loss functions and developing efficient optimization algorithms. This paper aims to extend results for general nonlinear embeddings. First, we finish detailed derivations and provide clean formulations for efficiently calculating some building blocks of optimization algorithms such as function, gradient evaluation, and Hessian-vector product. The result enables the use of many optimization methods for extreme similarity learning with nonlinear embeddings. Second, we study some optimization methods in detail. Due to the use of nonlinear embeddings, implementation issues different from linear cases are addressed. In the end, some methods are shown to be highly efficient for extreme similarity learning with nonlinear embeddings. Bo-Wen Yuan, Yu-Sheng Li 0001, Pengrui Quan, Chih-Jen Lin |
KDD | 4 |
| 2021 | A Two-Level Decomposition Framework Exploiting First and Second Order Information for SVM Training ProblemsabstractIn this work we present a novel way to solve the sub-problems that originate when using decomposition algorithms to train Support Vector Machines (SVMs). State-of-the-art Sequential Minimization Optimization (SMO) solvers reduce the original problem to a sequence of sub-problems of two variables for which the solution is analytical. Although considering more than two variables at a time usually results in a lower number of iterations needed to train an SVM model, solving the sub-problem becomes much harder and the overall computational gains are limited, if any. We propose to apply the two-variables decomposition method to solve the sub-problems themselves and experimentally show that it is a viable and efficient way to deal with sub-problems of up to 50 variables. As a second contribution we explore different ways to select the working set and its size, combining first-order and second-order working set selection rules together with a strategy for exploiting cached elements of the Hessian matrix. An extensive numerical comparison shows that the method performs considerably better than state-of-the-art software. Giulio Galvan, Matteo Lapucci, Chih-Jen Lin, Marco Sciandrone |
J. Mach. Learn. Res. | 3 |
| 2020 | AutoConjunction: Adaptive Model-based Feature Conjunction for CTR PredictionabstractClick-through rate (CTR) prediction is an important topic in mobile recommendation systems and computational advertising. As previous research indicates, a key point to maximize CTR is feature conjunction for making the training data more informative. Despite great progress, existing methods still fail to choose suitable settings of feature conjunction for the given data. In particular, a linear model on the pair-wise feature conjunction may overfit the training set if the data set is highly sparse. For such data, a model based on low-rank latent matrices are shown to be more appropriate. Unfortunately, practitioners now face difficulties to decide when to use which. In this paper, we propose an adaptive framework to address the feature conjunction problem. Our proposed framework adaptively chooses effective models to do feature conjunction according to data properties. We offer a case for building feature conjunction based on feature-pair frequency. Efficient training, as well as parameter selection, are thoroughly investigated. We conduct comprehensive online and offline experiments to demonstrate the effectiveness of the adaptive model over existing models for CTR prediction. Chih-Yao Chang, Xing Tang 0007, Bo-Wen Yuan, Jui-Yang Hsia, Zhenhua Dong, Xiuqiang He 0001, Chih-Jen Lin |
MDM | 8 |
| 2020 | Unbiased Ad Click Prediction for Position-aware Advertising SystemsabstractClick-through rate (CTR) prediction is a core problem of building advertising systems. In many real-world applications, because an ad placed in various positions has different click probabilities, the position information should be considered in both training and prediction. For such position-aware systems, existing approaches learn CTR models from clicks/not-clicks on historically displayed events by leveraging the position information in different ways. In this work, we explain that these approaches may give a heavily biased model. We first point out that in position-aware systems, two different types of selection biases coexist in displayed events. Secondly, we explain that some approaches attempting to eliminate the position effect from clicks/not-clicks may possess an additional bias. Finally, to obtain an unbiased CTR model for position-aware systems, we propose a novel counterfactual learning framework. Experiments confirm both our analysis on selection biases and the effectiveness of our proposed counterfactual learning framework. Bo-Wen Yuan, Yaxu Liu, Jui-Yang Hsia, Zhenhua Dong, Chih-Jen Lin |
RecSys | 5 |
| 2020 | Two-variable Dual Coordinate Descent Methods for Linear SVM with/without the Bias TermabstractCoordinate descent (CD) methods have been a state-of-the-art technique for training large-scale linear SVM. The most used setting is to solve the dual problem of an SVM formulation without the bias term (or an SVM formulation by embedding the bias term in the weight vector). The reason of omitting the bias term is that dual SVM no longer has a linear constraint and the CD procedure of updating one variable at a time is very simple. However, some have criticized the decision of not considering the bias term. To understand the role of the bias term in the design of CD methods for linear SVM, we give a thorough study on two-variable CD. First, if the bias term is not considered, we develop a two-variable CD that is competitive with the commonly used one-variable CD and is superior for difficult problems. The procedure is simple and has theoretical linear-rate convergence. Second, we investigate two-variable CD for linear SVM with the bias term. Analysis shows that CD is much less efficient for such a setting. Therefore, we conclude that in using CD for linear SVM, in general the bias term should not be considered. Chi-Cheng Chiu, Pin-Yen Lin, Chih-Jen Lin |
SDM | 3 |
| 2020 | Dual Coordinate-Descent Methods for Linear One-Class SVM and SVDDabstractOne-class support vector machines (SVM) and support vector data description (SVDD) are two effective outlier detection techniques. They have been successfully applied to many applications under the kernel settings, but for some high dimensional data, linear rather than kernel one-class SVM and SVDD may be more suitable. Past developments on kernel and linear classification have indicated that specially designed optimization algorithms can make the training for linear scenarios much faster. However, we point out that because of some differences from standard linear SVM, existing algorithms may not be efficient for one-class scenarios. We then develop some novel coordinate descent methods for linear one-class SVM and SVDD. Experiments demonstrate their superiority on the convergence speed. Hung-Yi Chou, Pin-Yen Lin, Chih-Jen Lin |
SDM | 3 |
| 2020 | Newton Methods for Convolutional Neural NetworksabstractDeep learning involves a difficult non-convex optimization problem, which is often solved by stochastic gradient (SG) methods. While SG is usually effective, it may not be robust in some situations. Recently, Newton methods have been investigated as an alternative optimization technique, but most existing studies consider only fully connected feedforward neural networks. These studies do not investigate some more commonly used networks such as Convolutional Neural Networks (CNN). One reason is that Newton methods for CNN involve complicated operations, and so far no works have conducted a thorough investigation. In this work, we give details of all building blocks, including the evaluation of function, gradient, Jacobian, and Gauss-Newton matrix-vector products. These basic components are very important not only for practical implementation but also for developing variants of Newton methods for CNN. We show that an efficient MATLAB implementation can be done in just several hundred lines of code. Preliminary experiments indicate that Newton methods are less sensitive to parameters than the stochastic gradient approach. Chien-Chih Wang, Kent Loong Tan, Chih-Jen Lin |
ACM Trans. Intell. Syst. Technol. | 3 |
| 2020 | Introduction to the Special Issue on the Best Papers from KDD 2018abstractNo abstract available. Chih-Jen Lin |
ACM Trans. Knowl. Discov. Data | 2 |
| 2020 | Parameter Selection for Linear Support Vector RegressionabstractIn linear support vector regression (SVR), the regularization and error sensitivity parameters are used to avoid overfitting the training data. A proper selection of parameters is very essential for obtaining a good model, but the search process may be complicated and time-consuming. In an earlier work by Chu et al. (2015), an effective parameter-selection procedure by using warm-start techniques to solve a sequence of optimization problems has been proposed for linear classification. We extend their techniques to linear SVR, but address some new and challenging issues. In particular, linear classification involves only the regularization parameter, but linear SVR has an extra error sensitivity parameter. We investigate the effective range of each parameter and the sequence in checking the two parameters. Based on this work, an effective tool for the selection of parameters for linear SVR has been available for public use. Jui-Yang Hsia, Chih-Jen Lin |
IEEE Trans. Neural Networks Learn. Syst. | 2 |
| 2019 | Improving Ad Click Prediction by Considering Non-displayed EventsabstractClick-through rate (CTR) prediction is the core problem of building advertising systems. Most existing state-of-the-art approaches model CTR prediction as binary classification problems, where displayed events with and without click feedbacks are respectively considered as positive and negative instances for training and offline validation. However, due to the selection mechanism applied in most advertising systems, a selection bias exists between distributions of displayed and non-displayed events. Conventional CTR models ignoring the bias may have inaccurate predictions and cause a loss of the revenue. To alleviate the bias, we need to conduct counterfactual learning by considering not only displayed events but also non-displayed events. In this paper, through a review of existing approaches of counterfactual learning, we point out some difficulties for applying these approaches for CTR prediction in a real-world advertising system. To overcome these difficulties, we propose a novel framework for counterfactual CTR prediction. In experiments, we compare our proposed framework against state-of-the-art conventional CTR models and existing counterfactual learning approaches. Experimental results show significant improvements. Bo-Wen Yuan, Jui-Yang Hsia, Hong Zhu 0003, Chih-Yao Chang, Zhenhua Dong, Chih-Jen Lin |
CIKM | 7 |
| 2019 | The Common-directions Method for Regularized Empirical Risk MinimizationabstractState-of-the-art first- and second-order optimization methods are able to achieve either fast global linear convergence rates or quadratic convergence, but not both of them. In this work, we propose an interpolation between first- and second-order methods for regularized empirical risk minimization that exploits the problem structure to efficiently combine multiple update directions. Our method attains both optimal global linear convergence rate for first-order methods, and local quadratic convergence. Experimental results show that our method outperforms state-of-the-art first- and second-order optimization methods in terms of the number of data accesses, while is competitive in training time. Po-Wei Wang, Ching-Pei Lee, Chih-Jen Lin |
J. Mach. Learn. Res. | 3 |
| 2018 | Preconditioned Conjugate Gradient Methods in Truncated Newton Frameworks for Large-scale Linear ClassificationabstractTruncated Newton method is one of the most effective optimization methods for large-scale linear classification. The main computational task at each Newton iteration is to approximately solve a quadratic sub-problem by an iterative procedure such as the conjugate gradient (CG) method. It is known that CG has slow convergence if the sub-problem is ill-conditioned. Preconditioned CG (PCG) methods have been used to improve the convergence of the CG method, but it is difficult to find a preconditioner that performs well in most situations. Further, because Hessian-free optimization techniques are incorporated for handling large data, many existing preconditioners are not directly applicable. In this work, we detailedly study some preconditioners that have been considered in past works for linear classification. We show that these preconditioners may not help to improve the training speed in some cases. After some investigation, we propose simple and effective techniques to make the PCG method more robust in a truncated Newton framework. The idea is to avoid the situation when a preconditioner leads to a much worse condition number than when it is not applied. We provide theoretical justification. Through carefully designed experiments, we demonstrate that our method can effectively reduce the training time for large-scale problems. Chih-Yang Hsia, Wei-Lin Chiang, Chih-Jen Lin |
ACML | 3 |
| 2018 | Naive Parallelization of Coordinate Descent Methods and an Application on Multi-core L1-regularized ClassificationabstractIt is well known that a direct parallelization of sequential optimization methods (e.g., coordinate descent and stochastic gradient methods) is often not effective. The reason is that at each iteration, the number of operations may be too small. We point out that this common understanding may not be true if the algorithm sequentially accesses the data in a feature-wise manner. For almost all real-world sparse sets we have examined, some features are much denser than others. Thus a direct parallelization of loops in a sequential method may result in excellent speedup. This approach possesses an advantage of retaining all convergence results because the algorithm is not changed at all. We apply this idea on coordinate descent (CD) methods, which are effective single-thread technique for L1-regularized classification. Further, an investigation on the shrinking technique commonly used to remove some features in the training process shows that this technique helps the parallelization of CD methods. Experiments indicate that a naive parallelization achieves better speedup than existing methods that laboriously modify the algorithm to achieve parallelism. Though a bit ironic, we conclude that the naive parallelization of the CD method is a highly competitive and robust multi-core implementation for L1-regularized classification. Yong Zhuang, Yu-Chin Juan, Guo-Xun Yuan, Chih-Jen Lin |
CIKM | 4 |
| 2018 | Limited-memory Common-directions Method for Distributed Ll-regularized Linear ClassificationabstractFor distributed linear classification, L1 regularization is useful because of a smaller model size. However, with the non-differentiability, it is more difficult to develop efficient optimization algorithms. In the past decade, OWLQN has emerged as the major method for distributed training of L1 problems. In this work, we point out issues in OWLQN's search directions. Then we extend the recently developed limited-memory common-directions method for L2-regularized problems to L1 scenarios. Through a unified interpretation of batch methods for L1 problems, we explain why OWLQN has been a popular method and why our method is superior in distributed environments. Experiments confirm that the proposed method is faster than OWLQN in most situations. Wei-Lin Chiang, Yu-Sheng Li 0001, Ching-Pei Lee, Chih-Jen Lin |
SDM | 4 |
| 2018 | Distributed Newton Methods for Deep Neural NetworksabstractDeep learning involves a difficult nonconvex optimization problem with a large number of weights between any two adjacent layers of a deep structure. To handle large data sets or complicated networks, distributed training is needed, but the calculation of function, gradient, and Hessian is expensive. In particular, the communication and the synchronization cost may become a bottleneck. In this letter, we focus on situations where the model is distributedly stored and propose a novel distributed Newton method for training deep neural networks. By variable and feature-wise data partitions and some careful designs, we are able to explicitly use the Jacobian matrix for matrix-vector products in the Newton method. Some techniques are incorporated to reduce the running time as well as memory consumption. First, to reduce the communication cost, we propose a diagonalization method such that an approximate Newton direction can be obtained without communication between machines. Second, we consider subsampled Gauss-Newton matrices for reducing the running time as well as the communication cost. Third, to reduce the synchronization cost, we terminate the process of finding an approximate Newton direction even though some nodes have not finished their tasks. Details of some implementation issues in distributed environments are thoroughly investigated. Experiments demonstrate that the proposed method is effective for the distributed training of deep neural networks. Compared with stochastic gradient methods, it is more robust and may give better test accuracy. Chien-Chih Wang, Kent Loong Tan, Chun-Ting Chen, Yu-Hsiang Lin, S. Sathiya Keerthi, Dhruv Mahajan 0001, S. Sundararajan, Chih-Jen Lin |
Neural Comput. | 8 |
| 2018 | An Efficient Alternating Newton Method for Learning Factorization MachinesabstractTo date, factorization machines (FMs) have emerged as a powerful model in many applications. In this work, we study the training of FM with the logistic loss for binary classification, which is a nonlinear extension of the linear model with the logistic loss (i.e., logistic regression). For the training of large-scale logistic regression, Newton methods have been shown to be an effective approach, but it is difficult to apply such methods to FM because of the nonconvexity. We consider a modification of FM that is multiblock convex and propose an alternating minimization algorithm based on Newton methods. Some novel optimization techniques are introduced to reduce the running time. Our experiments demonstrate that the proposed algorithm is more efficient than stochastic gradient algorithms and coordinate descent methods. The parallelism of our method is also investigated for the acceleration in multithreading environments. Wei-Sheng Chin, Bo-Wen Yuan, Chih-Jen Lin |
ACM Trans. Intell. Syst. Technol. | 4 |
| 2017 | A Unified Algorithm for One-Cass Structured Matrix Factorization with Side InformationabstractIn many applications such as recommender systems and multi-label learning the task is to complete a partially observed binary matrix. Such PU learning (positive-unlabeled) problems can be solved by one-class matrix factorization (MF). In practice side information such as user or item features in recommender systems are often available besides the observed positive user-item connections. In this work we consider a generalization of one-class MF so that two types of side information are incorporated and a general convex loss function can be used. The resulting optimization problem is very challenging, but we derive an efficient and effective alternating minimization procedure. Experiments on large-scale multi-label learning and one-class recommender systems demonstrate the effectiveness of our proposed approach. Hsiang-Fu Yu, Hsin-Yuan Huang, Inderjit S. Dhillon, Chih-Jen Lin |
AAAI | 4 |
| 2017 | A Study on Trust Region Update Rules in Newton Methods for Large-scale Linear ClassificationabstractThe main task in training a linear classifier is to solve an unconstrained minimization problem. To apply an optimization method typically we iteratively find a good direction and then decide a suitable step size. Past developments of extending optimization methods for large-scale linear classification focus on finding the direction, but little attention has been paid on adjusting the step size. In this work, we explain that inappropriate step-size adjustment may lead to serious slow convergence. Among the two major methods for step-size selection, line search and trust region, we focus on investigating the trust region methods. After presenting some detailed analysis, we develop novel and effective techniques to adjust the trust-region size. Experiments indicate that our new settings significantly outperform existing implementations for large-scale linear classification. Chih-Yang Hsia, Ya Zhu, Chih-Jen Lin |
ACML | 3 |
| 2017 | Limited-memory Common-directions Method for Distributed Optimization and its Application on Empirical Risk MinimizationabstractDistributed optimization has become an important research topic for dealing with extremely large volume of data available in the Internet companies nowadays. Additional machines make computation less expensive, but inter-machine communication becomes prominent in the optimization process, and efficient optimization methods should reduce the amount of the communication in order to achieve shorter overall running time. In this work, we utilize the advantages of the recently proposed, theoretically fast-convergent common-directions method, but tackle its main drawback of excessive spatial and computational costs to propose a limited-memory algorithm. The result is an efficient, linear-convergent optimization method for paraliel/distributed optimization. We further discuss how our method can exploit the problem structure to efficiently train regularized empirical risk minimization (ERM) models. Experimental results show that our method outperforms state-of-the-art distributed optimization methods for ERM problems. Ching-Pei Lee, Po-Wei Wang, Weizhu Chen, Chih-Jen Lin |
SDM | 4 |
| 2017 | Selection of Negative Samples for One-class Matrix FactorizationabstractMany recommender systems have only implicit user feedback. The two possible ratings are positive and negative, but only part of positive entries are observed. One-class matrix factorization (MF) is a popular approach for such scenarios by treating some missing entries as negative. Two major ways to select negative entries are by sub-sampling a set with similar size to that of observed positive entries or by including all missing entries as negative. They are referred to as “subsampled” and “full” approaches in this work, respectively. Currently detailed comparisons between these two selection schemes on large-scale data are still lacking. One important reason is that the “full” approach leads to a hard optimization problem after treating all missing entries as negative. In this paper, we successfully develop efficient optimization techniques to solve this challenging problem so that the “full” approach becomes practically viable. We then compare in detail the two approaches “subsampled” and “full” for selecting negative entries. Results show that the “full” approach of including much more missing entries as negative yields better results. Hsiang-Fu Yu, Mikhail Bilenko, Chih-Jen Lin |
SDM | 3 |
| 2016 | Parallel Dual Coordinate Descent Method for Large-scale Linear Classification in Multi-core EnvironmentsabstractDual coordinate descent method is one of the most effective approaches for large-scale linear classification. However, its sequential design makes the parallelization difficult. In this work, we target at the parallelization in a multi-core environment. After pointing out difficulties faced in some existing approaches, we propose a new framework to parallelize the dual coordinate descent method. The key idea is to make the majority of all operations (gradient calculation here) parallelizable. The proposed framework is shown to be theoretically sound. Further, we demonstrate through experiments that the new framework is robust and efficient in a multi-core environment. Wei-Lin Chiang, Mu-Chu Lee, Chih-Jen Lin |
KDD | 3 |
| 2016 | Field-aware Factorization Machines for CTR PredictionabstractClick-through rate (CTR) prediction plays an important role in computational advertising. Models based on degree-2 polynomial mappings and factorization machines (FMs) are widely used for this task. Recently, a variant of FMs, field-aware factorization machines (FFMs), outperforms existing models in some world-wide CTR-prediction competitions. Based on our experiences in winning two of them, in this paper we establish FFMs as an effective method for classifying large sparse data including those from CTR prediction. First, we propose efficient implementations for training FFMs. Then we comprehensively analyze FFMs and compare this approach with competing models. Experiments show that FFMs are very useful for certain classification problems. Finally, we have released a package of FFMs for public use. Yu-Chin Juan, Yong Zhuang, Wei-Sheng Chin, Chih-Jen Lin |
RecSys | 4 |
| 2016 | Linear and Kernel Classification: When to Use Which?abstractKernel methods are known to be a state-of-the-art classification technique. Nevertheless, the training and prediction cost is expensive for large data. On the other hand, linear classifiers can easily scale up, but are inferior to kernel classifiers in terms of predictability. Recent research has shown that for some data sets (e.g., document data), linear is as good as kernel classifiers. In such cases, the training of a kernel classifier is a waste of both time and memory. In this work, we investigate the important issue of efficiently and automatically deciding whether kernel classifiers perform strictly better than linear for a given data set. Our proposed method is based on cheaply constructing a classifier that exhibits nonlinearity and can be automatically trained. Then we make a decision by comparing the performance of our constructed classifier with the linear classifier. We propose two methods: the first one trains the degree-2 feature expansion by a linear-classification method, while the second dissects the feature space into several regions and trains a linear classifier for each region. The design considerations of our methods are very different from past works for speeding up the kernel training. They still aim at obtaining accuracy close to the kernel classifier, but ours would like to give a quick and accurate decision without worrying about accuracy. Empirically our methods can efficiently make correct indications for a wide variety of data sets. Our proposed process can thus be a useful component for automatic machine learning. Hsin-Yuan Huang, Chih-Jen Lin |
SDM | 2 |
| 2016 | LIBMF: A Library for Parallel Matrix Factorization in Shared-memory SystemsabstractMatrix factorization (MF) plays a key role in many applications such as recommender systems and computer vision, but MF may take long running time for handling large matrices commonly seen in the big data era. Many parallel techniques have been proposed to reduce the running time, but few parallel MF packages are available. Therefore, we present an open source library, LIBMF, based on recent advances of parallel MF for shared-memory systems. LIBMF includes easy-to-use command-line tools, interfaces to C/C++ languages, and comprehensive documentation. Our experiments demonstrate that LIBMF outperforms state of the art packages. LIBMF is BSD-licensed, so users can freely use, modify, and redistribute the code. Wei-Sheng Chin, Bo-Wen Yuan, Yong Zhuang, Yu-Chin Juan, Chih-Jen Lin |
J. Mach. Learn. Res. | 6 |
| 2015 | Fast Matrix-Vector Multiplications for Large-Scale Logistic Regression on Shared-Memory SystemsabstractShared-memory systems such as regular desktops now possess enough memory to store large data. However, the training process for data classification can still be slow if we do not fully utilize the power of multi-core CPUs. Many existing works proposed parallel machine learning algorithms by modifying serial ones, but convergence analysis may be complicated. Instead, we do not modify machine learning algorithms, but consider those that can take the advantage of parallel matrix operations. We particularly investigate the use of parallel sparse matrix-vector multiplications in a Newton method for large scale logistic regression. Various implementations from easy to sophisticated ones are analyzed and compared. Results indicate that under suitable settings excellent speedup can be achieved. Mu-Chu Lee, Wei-Lin Chiang, Chih-Jen Lin |
ICDM | 3 |
| 2015 | Warm Start for Parameter Selection of Linear ClassifiersabstractIn linear classification, a regularization term effectively remedies the overfitting problem, but selecting a good regularization parameter is usually time consuming. We consider cross validation for the selection process, so several optimization problems under different parameters must be solved. Our aim is to devise effective warm-start strategies to efficiently solve this sequence of optimization problems. We detailedly investigate the relationship between optimal solutions of logistic regression/linear SVM and regularization parameters. Based on the analysis, we develop an efficient tool to automatically find a suitable parameter for users with no related background knowledge. Bo-Yu Chu, Chia-Hua Ho, Cheng-Hao Tsai, Chieh-Yen Lin, Chih-Jen Lin |
KDD | 5 |
| 2015 | A Learning-Rate Schedule for Stochastic Gradient Methods to Matrix Factorization
Wei-Sheng Chin, Yong Zhuang, Yu-Chin Juan, Chih-Jen Lin |
PAKDD (1) | 4 |
| 2015 | Distributed Newton Methods for Regularized Logistic Regression
Yong Zhuang, Wei-Sheng Chin, Yu-Chin Juan, Chih-Jen Lin |
PAKDD (2) | 4 |
| 2015 | Combination of feature engineering and ranking models for paper-author identification in KDD cup 2013
Chun-Liang Li, Yu-Chuan Su, Ting-Wei Lin, Cheng-Hao Tsai, Wei-Cheng Chang, Kuan-Hao Huang, Tzu-Ming Kuo, Shan-Wei Lin, Young-San Lin, Yu-Chen Lu, Chun-Pai Yang, Cheng-Xia Chang, Wei-Sheng Chin, Yu-Chin Juan, Hsiao-Yu Fish Tung, Jui-Pin Wang, Cheng-Kuang Wei, Felix Wu, Tu-Chun Yin, Tong Yu 0001, Yong Zhuang, Shou-De Lin, Hsuan-Tien Lin, Chih-Jen Lin |
J. Mach. Learn. Res. | 24 |
| 2015 | Subsampled Hessian Newton Methods for Supervised LearningabstractNewton methods can be applied in many supervised learning approaches. However, for large-scale data, the use of the whole Hessian matrix can be time-consuming. Recently, subsampled Newton methods have been proposed to reduce the computational time by using only a subset of data for calculating an approximation of the Hessian matrix. Unfortunately, we find that in some situations, the running speed is worse than the standard Newton method because cheaper but less accurate search directions are used. In this work, we propose some novel techniques to improve the existing subsampled Hessian Newton method. The main idea is to solve a two-dimensional subproblem per iteration to adjust the search direction to better minimize the second-order approximation of the function value. We prove the theoretical convergence of the proposed method. Experiments on logistic regression, linear SVM, maximum entropy, and deep networks indicate that our techniques significantly reduce the running time of the subsampled Hessian Newton method. The resulting algorithm becomes a compelling alternative to the standard Newton method for large-scale data classification. Chien-Chih Wang, Chun-Heng Huang, Chih-Jen Lin |
Neural Comput. | 3 |
| 2015 | A Fast Parallel Stochastic Gradient Method for Matrix Factorization in Shared Memory SystemsabstractMatrix factorization is known to be an effective method for recommender systems that are given only the ratings from users to items. Currently, stochastic gradient (SG) method is one of the most popular algorithms for matrix factorization. However, as a sequential approach, SG is difficult to be parallelized for handling web-scale problems. In this article, we develop a fast parallel SG method, FPSG, for shared memory systems. By dramatically reducing the cache-miss rate and carefully addressing the load balance of threads, FPSG is more efficient than state-of-the-art parallel algorithms for matrix factorization. Wei-Sheng Chin, Yong Zhuang, Yu-Chin Juan, Chih-Jen Lin |
ACM Trans. Intell. Syst. Technol. | 4 |
| 2014 | Large-scale logistic regression and linear support vector machines using sparkabstractLogistic regression and linear SVM are useful methods for large-scale classification. However, their distributed implementations have not been well studied. Recently, because of the inefficiency of the MapReduce framework on iterative algorithms, Spark, an in-memory cluster-computing platform, has been proposed. It has emerged as a popular framework for large-scale data processing and analytics. In this work, we consider a distributed Newton method for solving logistic regression as well linear SVM and implement it on Spark. We carefully examine many implementation issues significantly affecting the running time and propose our solutions. After conducting thorough empirical investigations, we release an efficient and easy-to-use tool for the Spark community. Chieh-Yen Lin, Cheng-Hao Tsai, Ching-Pei Lee, Chih-Jen Lin |
IEEE BigData | 4 |
| 2014 | Incremental and decremental training for linear classificationabstractIn classification, if a small number of instances is added or removed, incremental and decremental techniques can be applied to quickly update the model. However, the design of incremental and decremental algorithms involves many considerations. In this paper, we focus on linear classifiers including logistic regression and linear SVM because of their simplicity over kernel or other methods. By applying a warm start strategy, we investigate issues such as using primal or dual formulation, choosing optimization methods, and creating practical implementations. Through theoretical analysis and practical experiments, we conclude that a warm start setting on a high-order optimization method for primal formulations is more suitable than others for incremental and decremental learning of linear classification. Cheng-Hao Tsai, Chieh-Yen Lin, Chih-Jen Lin |
KDD | 3 |
| 2014 | Large-scale Kernel RankSVMabstractLearning to rank is an important task for recommendation systems, online advertisement and web search. Among those learning to rank methods, rankSVM is a widely used model. Both linear and nonlinear (kernel) rankSVM have been extensively studied, but the lengthy training time of kernel rankSVM remains a challenging issue. In this paper, after discussing difficulties of training kernel rankSVM, we propose an efficient method to handle these problems. The idea is to reduce the number of variables from quadratic to linear with respect to the number of training instances, and efficiently evaluate the pairwise losses. Our setting is applicable to a variety of loss functions. Further, general optimization methods can be easily applied to solve the reformulated problem. Implementation issues are also carefully considered. Experiments show that our method is faster than state-of-the-art methods for training kernel rankSVM. Tzu-Ming Kuo, Ching-Pei Lee, Chih-Jen Lin |
SDM | 3 |
| 2014 | Effective string processing and matching for author disambiguation
Wei-Sheng Chin, Yong Zhuang, Yu-Chin Juan, Felix Wu, Hsiao-Yu Fish Tung, Tong Yu 0001, Jui-Pin Wang, Cheng-Xia Chang, Chun-Pai Yang, Wei-Cheng Chang, Kuan-Hao Huang, Tzu-Ming Kuo, Shan-Wei Lin, Young-San Lin, Yu-Chen Lu, Yu-Chuan Su, Cheng-Kuang Wei, Tu-Chun Yin, Chun-Liang Li, Ting-Wei Lin, Cheng-Hao Tsai, Shou-De Lin, Hsuan-Tien Lin, Chih-Jen Lin |
J. Mach. Learn. Res. | 24 |
| 2014 | Iteration complexity of feasible descent methods for convex optimization
Po-Wei Wang, Chih-Jen Lin |
J. Mach. Learn. Res. | 2 |
| 2014 | Large-Scale Linear RankSVMabstractLinear rankSVM is one of the widely used methods for learning to rank. Although its performance may be inferior to nonlinear methods such as kernel rankSVM and gradient boosting decision trees, linear rankSVM is useful to quickly produce a baseline model. Furthermore, following its recent development for classification, linear rankSVM may give competitive performance for large and sparse data. A great deal of works have studied linear rankSVM. The focus is on the computational efficiency when the number of preference pairs is large. In this letter, we systematically study existing works, discuss their advantages and disadvantages, and propose an efficient algorithm. We discuss different implementation issues and extensions with detailed experiments. Finally, we develop a robust linear rankSVM tool for public use. Ching-Pei Lee, Chih-Jen Lin |
Neural Comput. | 2 |
| 2014 | Big Data Small Footprint: The Design of A Low-Power Classifier for Detecting Transportation ModesabstractSensors on mobile phones and wearables, and in general sensors on IoT (Internet of Things), bring forth a couple of new challenges to big data research. First, the power consumption for analyzing sensor data must be low, since most wearables and portable devices are power-strapped. Second, the velocity of analyzing big data on these devices must be high, otherwise the limited local storage may overflow. This paper presents our hardware-software co-design of a classifier for wearables to detect a person's transportation mode (i.e., still, walking, running, biking, and on a vehicle). We particularly focus on addressing the big-data small-footprint requirement by designing a classifier that is low in both computational complexity and memory requirement. Together with a sensor-hub configuration, we are able to drastically reduce power consumption by 99%, while maintaining competitive mode-detection accuracy. The data used in the paper is made publicly available for conducting research. Meng-Chieh Yu, Tong Yu 0001, Shao-Chen Wang, Chih-Jen Lin, Edward Y. Chang |
Proc. VLDB Endow. | 4 |
| 2013 | Dense Non-rigid Point-Matching Using Random ProjectionsabstractWe present a robust and efficient technique for matching dense sets of points undergoing non-rigid spatial transformations. Our main intuition is that the subset of points that can be matched with high confidence should be used to guide the matching procedure for the rest. We propose a novel algorithm that incorporates these high-confidence matches as a spatial prior to learn a discriminative subspace that simultaneously encodes both the feature similarity as well as their spatial arrangement. Conventional subspace learning usually requires spectral decomposition of the pair-wise distance matrix across the point-sets, which can become inefficient even for moderately sized problems. To this end, we propose the use of random projections for approximate subspace learning, which can provide significant time improvements at the cost of minimal precision loss. This efficiency gain allows us to iteratively find and remove high-confidence matches from the point sets, resulting in high recall. To show the effectiveness of our approach, we present a systematic set of experiments and results for the problem of dense non-rigid image-feature matching. Raffay Hamid, Dennis DeCoste, Chih-Jen Lin |
CVPR | 3 |
| 2013 | Large-Scale Video Summarization Using Web-Image PriorsabstractGiven the enormous growth in user-generated videos, it is becoming increasingly important to be able to navigate them efficiently. As these videos are generally of poor quality, summarization methods designed for well-produced videos do not generalize to them. To address this challenge, we propose to use web-images as a prior to facilitate summarization of user-generated videos. Our main intuition is that people tend to take pictures of objects to capture them in a maximally informative way. Such images could therefore be used as prior information to summarize videos containing a similar set of objects. In this work, we apply our novel insight to develop a summarization algorithm that uses the web-image based prior information in an unsupervised manner. Moreover, to automatically evaluate summarization algorithms on a large scale, we propose a framework that relies on multiple summaries obtained through crowdsourcing. We demonstrate the effectiveness of our evaluation framework by comparing its performance to that of multiple human evaluators. Finally, we present results for our framework tested on hundreds of user-generated videos. Aditya Khosla, Raffay Hamid, Chih-Jen Lin, Neel Sundaresan |
CVPR | 3 |
| 2013 | A fast parallel SGD for matrix factorization in shared memory systemsabstractMatrix factorization is known to be an effective method for recommender systems that are given only the ratings from users to items. Currently, stochastic gradient descent (SGD) is one of the most popular algorithms for matrix factorization. However, as a sequential approach, SGD is difficult to be parallelized for handling web-scale problems. In this paper, we develop a fast parallel SGD method, FPSGD, for shared memory systems. By dramatically reducing the cache-miss rate and carefully addressing the load balance of threads, FPSGD is more efficient than state-of-the-art parallel algorithms for matrix factorization. Yong Zhuang, Wei-Sheng Chin, Yu-Chin Juan, Chih-Jen Lin |
RecSys | 4 |
| 2013 | A Study on L2-Loss (Squared Hinge-Loss) Multiclass SVMabstractCrammer and Singer's method is one of the most popular multiclass support vector machines (SVMs). It considers L1 loss (hinge loss) in a complicated optimization problem. In SVM, squared hinge loss (L2 loss) is a common alternative to L1 loss, but surprisingly we have not seen any paper studying the details of Crammer and Singer's method using L2 loss. In this letter, we conduct a thorough investigation. We show that the derivation is not trivial and has some subtle differences from the L1 case. Details provided in this work can be a useful reference for those who intend to use Crammer and Singer's method with L2 loss. They do not need a tedious process to derive everything by themselves. Furthermore, we present some new results on and discussion of both L1- and L2-loss formulations. Ching-Pei Lee, Chih-Jen Lin |
Neural Comput. | 2 |
| 2012 | Experiences and lessons in developing industry-strength machine learning and data mining softwareabstractTraditionally academic machine learning and data mining researchers focus on proposing new algorithms. The task of implementing these methods is often left to companies that are developing software packages. However, the gap between the two sides has caused some problems. First, the practical deployment of new algorithms still involves some challenging issues that need to be studied by researchers. Second, without further investigation after publishing their papers, researchers have neither the opportunity to work with real problems nor see how their methods are used. We discuss the experiences in developing two machine learning packages LIBSVM and LIBLINEAR, that are widely used in both academia and industry. We demonstrate that the interaction with users leads us to identify some important research problems. For example, the decision to study and then support multi-class SVM was essential in the early stage of developing LIBSVM. The birth of LIBLINEAR was driven by the need to classify large-scale documents in Internet companies. For fast training of large-scale problems, we had to create new algorithms other than those used in LIBSVM for kernel SVM. We present some practical use of LIBLINEAR for Internet applications. Finally, we give lessons learned and future perspectives for developing industry-strength machine learning and data mining software. Chih-Jen Lin |
KDD | 1 |
| 2012 | Large-scale linear support vector regression
Chia-Hua Ho, Chih-Jen Lin |
J. Mach. Learn. Res. | 2 |
| 2012 | An Improved GLMNET for L1-regularized Logistic Regression
Guo-Xun Yuan, Chia-Hua Ho, Chih-Jen Lin |
J. Mach. Learn. Res. | 3 |
| 2012 | Recent Advances of Large-Scale Linear ClassificationabstractLinear classification is a useful tool in machine learning and data mining. For some data in a rich dimensional space, the performance (i.e., testing accuracy) of linear classifiers has shown to be close to that of nonlinear classifiers such as kernel methods, but training and testing speed is much faster. Recently, many research works have developed efficient optimization methods to construct linear classifiers and applied them to some large-scale applications. In this paper, we give a comprehensive survey on the recent development of this active research area. Guo-Xun Yuan, Chia-Hua Ho, Chih-Jen Lin |
Proc. IEEE | 3 |
| 2012 | Large Linear Classification When Data Cannot Fit in MemoryabstractRecent advances in linear classification have shown that for applications such as document classification, the training process can be extremely efficient. However, most of the existing training methods are designed by assuming that data can be stored in the computer memory. These methods cannot be easily applied to data larger than the memory capacity due to the random access to the disk. We propose and analyze a block minimization framework for data larger than the memory size. At each step a block of data is loaded from the disk and handled by certain learning methods. We investigate two implementations of the proposed framework for primal and dual SVMs, respectively. Because data cannot fit in memory, many design considerations are very different from those for traditional algorithms. We discuss and compare with existing approaches that are able to handle data larger than memory. Experiments using data sets 20 times larger than the memory demonstrate the effectiveness of the proposed method. Hsiang-Fu Yu, Cho-Jui Hsieh, Kai-Wei Chang 0001, Chih-Jen Lin |
ACM Trans. Knowl. Discov. Data | 4 |
| 2011 | Large Linear Classification When Data Cannot Fit in MemoryabstractLinear classification is a useful tool for dealing with large-scale data in applications such as document classification and natural language processing. Recent developments of linear classification have shown that the training process can be efficiently conducted. However, when the data size exceeds the memory capacity, most training methods suffer from very slow convergence due to the severe disk swapping. Although some methods have attempted to handle such a situation, they are usually too complicated to support some important functions such as parameter selection. In this paper, we introduce a block minimization framework for data larger than memory. Under the framework, a solver splits data into blocks and stores them into separate files. Then, at each time, the solver trains a data block loaded from disk. Although the framework is simple, the experimental results show that it effectively handles a data set 20 times larger than the memory capacity. Hsiang-Fu Yu, Cho-Jui Hsieh, Kai-Wei Chang 0001, Chih-Jen Lin |
IJCAI | 4 |
| 2011 | An improved GLMNET for l1-regularized logistic regressionabstractGLMNET proposed by Friedman et al. is an algorithm for generalized linear models with elastic net. It has been widely applied to solve L1-regularized logistic regression. However, recent experiments indicated that the existing GLMNET implementation may not be stable for large-scale problems. In this paper, we propose an improved GLMNET to address some theoretical and implementation issues. In particular, as a Newton-type method, GLMNET achieves fast local convergence, but may fail to quickly obtain a useful solution. By a careful design to adjust the effort for each iteration, our method is efficient regardless of loosely or strictly solving the optimization problem. Experiments demonstrate that the improved GLMNET is more efficient than a state-of-the-art coordinate descent method. Guo-Xun Yuan, Chia-Hua Ho, Chih-Jen Lin |
KDD | 3 |
| 2011 | A Bayesian Approximation Method for Online Ranking
Ruby C. Weng, Chih-Jen Lin |
J. Mach. Learn. Res. | 2 |
| 2011 | Dual coordinate descent methods for logistic regression and maximum entropy models
Hsiang-Fu Yu, Fang-Lan Huang, Chih-Jen Lin |
Mach. Learn. | 3 |
| 2011 | Parallel Spectral Clustering in Distributed SystemsabstractSpectral clustering algorithms have been shown to be more effective in finding clusters than some traditional algorithms, such as k-means. However, spectral clustering suffers from a scalability problem in both memory use and computational time when the size of a data set is large. To perform clustering on large data sets, we investigate two representative ways of approximating the dense similarity matrix. We compare one approach by sparsifying the matrix with another by the Nyström method. We then pick the strategy of sparsifying the matrix via retaining nearest neighbors and investigate its parallelization. We parallelize both memory use and computation on distributed computers. Through an empirical study on a document data set of 193,844 instances and a photo data set of 2,121,863, we show that our parallel algorithm can effectively handle large problems. Wen-Yen Chen, Yangqiu Song, Hongjie Bai, Chih-Jen Lin, Edward Y. Chang |
IEEE Trans. Pattern Anal. Mach. Intell. | 4 |
| 2011 | LIBSVM: A library for support vector machinesabstractLIBSVM is a library for Support Vector Machines (SVMs). We have been actively developing this package since the year 2000. The goal is to help users to easily apply SVM to their applications. LIBSVM has gained wide popularity in machine learning and many other areas. In this article, we present all implementation details of LIBSVM. Issues such as solving SVM optimization problems theoretical convergence multiclass classification probability estimates and parameter selection are discussed in detail. Chih-Chung Chang, Chih-Jen Lin |
ACM Trans. Intell. Syst. Technol. | 2 |
| 2010 | Designing, Analyzing and Exploiting Stake-Based Social NetworksabstractIt is widely recognized that stakeholder information can provide important knowledge about stock investments, and an increasing number of countries require that such information is publicly available. In this paper, we present a novel way to exploit stakeholder information by using it to construct stake-based social networks, namely, StakeNet. We also provide a visualization tool that displays socio-centric and ego-centric views of the networks. In addition, we analyze stakeholders' static and dynamic behavior patterns in StakeNet, and demonstrate that most of StakeNet's properties are similar to those of a typical social network, except that the in-degree distribution does not follow a power law distribution. Finally, we demonstrate two applications of StakeNet by exploiting it to identify important companies and to group companies together. The experiments show that our results are highly consistent with the outcomes generated by human experts. Source code, dataset, and resources are available at http://www.csie.ntu.edu.tw/~d97944007/stakenet/. Tsung-Ting Kuo, Jung-Jung Yeh, Chih-Jen Lin, Shou-De Lin |
ASONAM | 3 |
| 2010 | Active learning strategies using SVMsabstractIn this paper, we decompose the problem of active learning into two parts, learning with few examples and learning by querying labels of samples. The first part is achieved mainly by SVM classifiers. We also consider variants based on transductive learning. In the second part, based on SVM decision values, we propose a framework to flexibly select points for query. Our experiments are conducted on the data sets of Causality Active Learning Challenge. With measurements of Area Under Curve (AUC) and Area under the Learning Curve (ALC), we find suitable methods for different data sets. Ming-Hen Tsai, Chia-Hua Ho, Chih-Jen Lin |
IJCNN | 3 |
| 2010 | Large linear classification when data cannot fit in memoryabstractRecent advances in linear classification have shown that for applications such as document classification, the training can be extremely efficient. However, most of the existing training methods are designed by assuming that data can be stored in the computer memory. These methods cannot be easily applied to data larger than the memory capacity due to the random access to the disk. We propose and analyze a block minimization framework for data larger than the memory size. At each step a block of data is loaded from the disk and handled by certain learning methods. We investigate two implementations of the proposed framework for primal and dual SVMs, respectively. As data cannot fit in memory, many design considerations are very different from those for traditional algorithms. Experiments using data sets 20 times larger than the memory demonstrate the effectiveness of the proposed method. Hsiang-Fu Yu, Cho-Jui Hsieh, Kai-Wei Chang 0001, Chih-Jen Lin |
KDD | 4 |
| 2010 | Training and Testing Low-degree Polynomial Data Mappings via Linear SVM
Yin-Wen Chang, Cho-Jui Hsieh, Kai-Wei Chang 0001, Michael Ringgaard, Chih-Jen Lin |
J. Mach. Learn. Res. | 5 |
| 2010 | Iterative Scaling and Coordinate Descent Methods for Maximum Entropy Models
Fang-Lan Huang, Cho-Jui Hsieh, Kai-Wei Chang 0001, Chih-Jen Lin |
J. Mach. Learn. Res. | 4 |
| 2010 | A Comparison of Optimization Methods and Software for Large-scale L1-regularized Linear Classification
Guo-Xun Yuan, Kai-Wei Chang 0001, Cho-Jui Hsieh, Chih-Jen Lin |
J. Mach. Learn. Res. | 4 |
| 2008 | A dual coordinate descent method for large-scale linear SVMabstractIn many applications, data appear with a huge number of instances as well as features. Linear Support Vector Machines (SVM) is one of the most popular tools to deal with such large-scale sparse data. This paper presents a novel dual coordinate descent method for linear SVM with L1-and L2-loss functions. The proposed method is simple and reaches an ε-accurate solution in O(log(1/ε)) iterations. Experiments indicate that our method is much faster than state of the art solvers such as Pegasos, TRON, SVMperf, and a recent primal coordinate descent implementation. Cho-Jui Hsieh, Kai-Wei Chang 0001, Chih-Jen Lin, S. Sathiya Keerthi, S. Sundararajan |
ICML | 3 |
| 2008 | A sequential dual method for large scale multi-class linear svmsabstractEfficient training of direct multi-class formulations of linear Support Vector Machines is very useful in applications such as text classification with a huge number examples as well as features. This paper presents a fast dual method for this training. The main idea is to sequentially traverse through the training set and optimize the dual variables associated with one example at a time. The speed of training is enhanced further by shrinking and cooling heuristics. Experiments indicate that our method is much faster than state of the art solvers such as bundle, cutting plane and exponentiated gradient methods. S. Sathiya Keerthi, S. Sundararajan, Kai-Wei Chang 0001, Cho-Jui Hsieh, Chih-Jen Lin |
KDD | 5 |
| 2008 | Parallel Spectral Clustering
Yangqiu Song, Wen-Yen Chen, Hongjie Bai, Chih-Jen Lin, Edward Y. Chang |
ECML/PKDD (2) | 4 |
| 2008 | Cross-generation and cross-laboratory predictions of Affymetrix microarrays by rank-based methods
Hsi-Che Liu, Chien-Yu Chen 0001, Yu-Ting Liu, Cheng-Bang Chu, Der-Cherng Liang, Lee-Yung Shih, Chih-Jen Lin |
J. Biomed. Informatics | 7 |
| 2008 | Coordinate Descent Method for Large-scale L2-loss Linear Support Vector Machines
Kai-Wei Chang 0001, Cho-Jui Hsieh, Chih-Jen Lin |
J. Mach. Learn. Res. | 3 |
| 2008 | LIBLINEAR: A Library for Large Linear Classification
Rong-En Fan, Kai-Wei Chang 0001, Cho-Jui Hsieh, Xiang-Rui Wang, Chih-Jen Lin |
J. Mach. Learn. Res. | 5 |
| 2008 | Trust Region Newton Method for Logistic Regression
Chih-Jen Lin, Ruby C. Weng, S. Sathiya Keerthi |
J. Mach. Learn. Res. | 1 |
| 2007 | Trust region Newton methods for large-scale logistic regressionabstractLarge-scale logistic regression arises in many applications such as document classification and natural language processing. In this paper, we apply a trust region Newton method to maximize the log-likelihood of the logistic regression model. The proposed method uses only approximate Newton steps in the beginning, but achieves fast convergence in the end. Experiments show that it is faster than the commonly used quasi Newton approach for logistic regression. We also compare it with linear SVM implementations. Chih-Jen Lin, Ruby C. Weng, S. Sathiya Keerthi |
ICML | 1 |
| 2007 | A note on Platt's probabilistic outputs for support vector machines
Hsuan-Tien Lin, Chih-Jen Lin, Ruby C. Weng |
Mach. Learn. | 2 |
| 2007 | Projected Gradient Methods for Nonnegative Matrix FactorizationabstractNonnegative matrix factorization (NMF) can be formulated as a minimization problem with bound constraints. Although bound-constrained optimization has been studied extensively in both theory and practice, so far no study has formally applied its techniques to NMF. In this letter, we propose two projected gradient methods for NMF, both of which exhibit strong optimization properties. We discuss efficient implementations and demonstrate that one of the proposed methods converges faster than the popular multiplicative update approach. A simple Matlab code is also provided. Chih-Jen Lin |
Neural Comput. | 1 |
| 2007 | On the Convergence of Multiplicative Update Algorithms for Nonnegative Matrix FactorizationabstractNonnegative matrix factorization (NMF) is useful to find basis information of nonnegative data. Currently, multiplicative updates are a simple and popular way to find the factorization. However, for the common NMF approach of minimizing the Euclidean distance between approximate and true values, no proof has shown that multiplicative updates converge to a stationary point of the NMF optimization problem. Stationarity is important as it is a necessary condition of a local minimum. This paper discusses the difficulty of proving the convergence. We propose slight modifications of existing updates and prove their convergence. Techniques invented in this paper may be applied to prove the convergence for other bound-constrained optimization problems. Chih-Jen Lin |
IEEE Trans. Neural Networks | 1 |
| 2006 | Ranking individuals by group comparisonsabstractThis paper proposes new approaches to rank individuals from their group competition results. Many real-world problems are of this type. For example, ranking players from team games is important in some sports. We propose an exponential model to solve such problems. To estimate individual rankings through the proposed model we introduce two convex minimization formulas with easy and efficient solution procedures. Experiments on real bridge records and multi-class classification demonstrate the viability of the proposed model. Tzu-Kuo Huang, Chih-Jen Lin, Ruby C. Weng |
ICML | 2 |
| 2006 | Generalized Bradley-Terry Models and Multi-Class Probability EstimatesabstractThe Bradley-Terry model for obtaining individual skill from paired comparisons has been popular in many areas. In machine learning, this model is related to multi-class probability estimates by coupling all pairwise classification results. Error correcting output codes (ECOC) are a general framework to decompose a multi-class problem to several binary problems. To obtain probability estimates under this framework, this paper introduces a generalized Bradley-Terry model in which paired individual comparisons are extended to paired team comparisons. We propose a simple algorithm with convergence proofs to solve the model and obtain individual skill. Experiments on synthetic and re al data demonstrate that the algorithm is useful for obtaining multi-class probability estimates. Moreover, we discuss four extensions of the proposed model: 1) weighted individual skill, 2) home-field advantage, 3) ties, and 4) comparisons with more than two teams. Tzu-Kuo Huang, Ruby C. Weng, Chih-Jen Lin |
J. Mach. Learn. Res. | 3 |
| 2006 | A study on SMO-type decomposition methods for support vector machinesabstractDecomposition methods are currently one of the major methods for training support vector machines. They vary mainly according to different working set selections. Existing implementations and analysis usually consider some specific selection rules. This paper studies sequential minimal optimization type decomposition methods under a general and flexible way of choosing the two-element working set. The main results include: 1) a simple asymptotic convergence proof, 2) a general explanation of the shrinking and caching techniques, and 3) the linear convergence of the methods. Extensions to some support vector machine variants are also discussed. Pai-Hsuen Chen, Rong-En Fan, Chih-Jen Lin |
IEEE Trans. Neural Networks | 3 |
| 2005 | Training Support Vector Machines via SMO-Type Decomposition Methods
Pai-Hsuen Chen, Rong-En Fan, Chih-Jen Lin |
ALT | 3 |
| 2005 | Training Support Vector Machines via SMO-Type Decomposition Methods
Pai-Hsuen Chen, Rong-En Fan, Chih-Jen Lin |
Discovery Science | 3 |
| 2005 | Working Set Selection Using Second Order Information for Training Support Vector MachinesabstractWorking set selection is an important step in decomposition methods for training support vector machines (SVMs). This paper develops a new technique for working set selection in SMO-type decomposition methods. It uses second order information to achieve fast convergence. Theoretical properties such as linear convergence are established. Experiments demonstrate that the proposed method is faster than existing selection methods using first order information. Rong-En Fan, Pai-Hsuen Chen, Chih-Jen Lin |
J. Mach. Learn. Res. | 3 |
| 2005 | Leave-One-Out Bounds for Support Vector Regression Model SelectionabstractMinimizing bounds of leave-one-out errors is an important and efficient approach for support vector machine (SVM) model selection. Past research focuses on their use for classification but not regression. In this letter, we derive various leave-one-out bounds for support vector regression (SVR) and discuss the difference from those for classification. Experiments demonstrate that the proposed bounds are competitive with Bayesian SVR for parameter selection. We also discuss the differentiability of leave-one-out bounds. Ming-Wei Chang, Chih-Jen Lin |
Neural Comput. | 2 |
| 2004 | A Generalized Bradley-Terry Model: From Group Competition to Individual SkillabstractThe Bradley-Terry model for paired comparison has been popular in many areas. We propose a generalized version in which paired individual comparisons are extended to paired team comparisons. We introduce a simple algorithm with convergence proofs to solve the model and obtain individual skill. A useful application to multi-class probability estimates using error-correcting codes is demonstrated. Tzu-Kuo Huang, Chih-Jen Lin, Ruby C. Weng |
NIPS | 2 |
| 2004 | Probability Estimates for Multi-class Classification by Pairwise Coupling
Ting-Fan Wu, Chih-Jen Lin, Ruby C. Weng |
J. Mach. Learn. Res. | 2 |
| 2004 | Decomposition Methods for Linear Support Vector MachinesabstractIn this letter, we show that decomposition methods with alpha seeding are extremely useful for solving a sequence of linear support vector machines (SVMs) with more data than attributes. This strategy is motivated by Keerthi and Lin (2003), who proved that for an SVM with data not linearly separable, after C is large enough, the dual solutions have the same free and bounded components. We explain why a direct use of decomposition methods for linear SVMs is sometimes very slow and then analyze why alpha seeding is much more effective for linear than nonlinear SVMs. We also conduct comparisons with other methods that are efficient for linear SVMs and demonstrate the effectiveness of alpha seeding techniques in model selection. Wei-Chun Kao, Kai-Min Chung, Chia-Liang Sun, Chih-Jen Lin |
Neural Comput. | 4 |
| 2004 | Analysis of switching dynamics with competing support vector machinesabstractWe present a framework for the unsupervised segmentation of switching dynamics using support vector machines. Following the architecture by Pawelzik et al., where annealed competing neural networks were used to segment a nonstationary time series, in this paper, we exploit the use of support vector machines, a well-known learning technique. First, a new formulation of support vector regression is proposed. Second, an expectation-maximization step is suggested to adaptively adjust the annealing parameter. Results indicate that the proposed approach is promising. Ming-Wei Chang, Chih-Jen Lin, Ruby Chiu-Hsing Weng |
IEEE Trans. Neural Networks | 2 |
| 2003 | Decomposition methods for linear support vector machinesabstractWe explain that decomposition methods, in particular, SMO-type algorithms, are not suitable for linear SVMs with more data than attributes. To remedy this difficulty, we consider a recent result by S.S. Keerthi and C.-J. Lin (see http://www.csie.ntu.edu.tw//spl sim/cjlin/papers/limit.ps.gz, 2002) that for an SVM which is not linearly separable, after C is large enough, the dual solutions are at similar faces. Motivated by this property, we show that alpha seeding is extremely useful for solving a sequence of linear SVMs. It largely reduces the number of decomposition iterations to the point that solving many linear SVMs requires less time than the original decomposition method for one single SVM. We also conduct comparisons with other methods which are efficient for linear SVMs, and demonstrate the effectiveness of the proposed approach for helping the model selection. Kai-Min Chung, Wei-Chun Kao, Tony Sun, Chih-Jen Lin |
ICASSP (4) | 4 |
| 2003 | Probability Estimates for Multi-Class Classification by Pairwise CouplingabstractPairwise coupling is a popular multi-class classification method that combines together all pairwise comparisons for each pair of classes. This paper presents two approaches for obtaining class probabilities. Both methods can be reduced to linear systems and are easy to implement. We show conceptually and experimentally that the proposed approaches are more stable than two existing popular methods: voting and [3]. Ting-Fan Wu, Chih-Jen Lin, Ruby C. Weng |
NIPS | 2 |
| 2003 | Special issue on support vector machines
Colin Campbell, Chih-Jen Lin, S. Sathiya Keerthi, V. David Sánchez A. |
Neurocomputing | 2 |
| 2003 | Radius Margin Bounds for Support Vector Machines with the RBF KernelabstractAn important approach for efficient support vector machine (SVM) model selection is to use differentiable bounds of the leave-one-out (loo) error. Past efforts focused on finding tight bounds of loo (e.g., radius margin bounds, span bounds). However, their practical viability is still not very satisfactory. Duan, Keerthi, and Poo (2003) showed that radius margin bound gives good prediction for L2-SVM, one of the cases we look at. In this letter, through analyses about why this bound performs well for L2-SVM, we show that finding a bound whose minima are in a region with small loo values may be more important than its tightness. Based on this principle, we propose modified radius margin bounds for L1-SVM (the other case) where the original bound is applicable only to the hard-margin case. Our modification for L1-SVM achieves comparable performance to L2-SVM. To study whether L1- or L2-SVM should be used, we analyze other properties, such as their differentiability, number of support vectors, and number of free support vectors. In this aspect, L1-SVM possesses the advantage of having fewer support vectors. Their implementations are also different, so we discuss related issues in detail. Kai-Min Chung, Wei-Chun Kao, Chia-Liang Sun, Li-Lun Wang, Chih-Jen Lin |
Neural Comput. | 5 |
| 2003 | Asymptotic Behaviors of Support Vector Machines with Gaussian KernelabstractSupport vector machines (SVMs) with the gaussian (RBF) kernel have been popular for practical use. Model selection in this class of SVMs involves two hyperparameters: the penalty parameter C and the kernel width sigma. This letter analyzes the behavior of the SVM classifier when these hyperparameters take very small or very large values. Our results help in understanding the hyperparameter space that leads to an efficient heuristic method of searching for hyperparameter values with small generalization errors. The analysis also indicates that if complete model selection using the gaussian kernel has been conducted, there is no need to consider linear SVM. S. Sathiya Keerthi, Chih-Jen Lin |
Neural Comput. | 2 |
| 2003 | A study on reduced support vector machinesabstractRecently the reduced support vector machine (RSVM) was proposed as an alternate of the standard SVM. Motivated by resolving the difficulty on handling large data sets using SVM with nonlinear kernels, it preselects a subset of data as support vectors and solves a smaller optimization problem. However, several issues of its practical use have not been fully discussed yet. For example, we do not know if it possesses comparable generalization ability as the standard SVM. In addition, we would like to see for how large problems RSVM outperforms SVM on training time. In this paper we show that the RSVM formulation is already in a form of linear SVM and discuss four RSVM implementations. Experiments indicate that in general the test accuracy of RSVM are a little lower than that of the standard SVM. In addition, for problems with up to tens of thousands of data, if the percentage of support vectors is not high, existing implementations for SVM is quite competitive on the training time. Thus, from this empirical study, RSVM will be mainly useful for either larger problems or those with many support vectors. Experiments in this paper also serve as comparisons of: 1) different implementations for linear SVM and 2) standard SVM using linear and quadratic cost functions. Kuan-Ming Lin, Chih-Jen Lin |
IEEE Trans. Neural Networks | 2 |
| 2002 | A Simple Decomposition Method for Support Vector Machines
Chih-Jen Lin |
Mach. Learn. | 2 |
| 2002 | Training v -Support Vector Regression: Theory and AlgorithmsabstractWe discuss the relation between epsilon-support vector regression (epsilon-SVR) and nu-support vector regression (nu-SVR). In particular, we focus on properties that are different from those of C-support vector classification (C-SVC) and nu-support vector classification (nu-SVC). We then discuss some issues that do not occur in the case of classification: the possible range of epsilon and the scaling of target values. A practical decomposition method for nu-SVR is implemented, and computational experiments are conducted. We show some interesting numerical observations specific to regression. Chih-Chung Chang, Chih-Jen Lin |
Neural Comput. | 2 |
| 2002 | A Note on the Decomposition Methods for Support Vector RegressionabstractThe dual formulation of support vector regression involves two closely related sets of variables. When the decomposition method is used, many existing approaches use pairs of indices from these two sets as the working set. Basically, they select a base set first and then expand it so all indices are pairs. This makes the implementation different from that for support vector classification. In addition, a larger optimization subproblem has to be solved in each iteration. We provide theoretical proofs and conduct experiments to show that using the base set as the working set leads to similar convergence (number of iterations). Therefore, by using a smaller working set while keeping a similar number of iterations, the program can be simpler and more efficient. Shuo-Peng Liao, Hsuan-Tien Lin, Chih-Jen Lin |
Neural Comput. | 3 |
| 2002 | A comparison of methods for multiclass support vector machinesabstractSupport vector machines (SVMs) were originally designed for binary classification. How to effectively extend it for multiclass classification is still an ongoing research issue. Several methods have been proposed where typically we construct a multiclass classifier by combining several binary classifiers. Some authors also proposed methods that consider all classes at once. As it is computationally more expensive to solve multiclass problems, comparisons of these methods using large-scale problems have not been seriously conducted. Especially for methods solving multiclass SVM in one step, a much larger optimization problem is required so up to now experiments are limited to small data sets. In this paper we give decomposition implementations for two such "all-together" methods. We then compare their performance with three methods based on binary classifications: "one-against-all," "one-against-one," and directed acyclic graph SVM (DAGSVM). Our experiments indicate that the "one-against-one" and DAG methods are more suitable for practical use than the other methods. Results also show that for large problems methods by considering all data at once in general need fewer support vectors. Chih-Jen Lin |
IEEE Trans. Neural Networks | 2 |
| 2002 | Asymptotic convergence of an SMO algorithm without any assumptionsabstractThe asymptotic convergence of C.-J. Lin (2001) can be applied to a modified SMO (sequential minimal optimization) algorithm by S.S. Keerthi et al. (2001) with some assumptions. The author shows that for this algorithm those assumptions are not necessary. Chih-Jen Lin |
IEEE Trans. Neural Networks | 1 |
| 2002 | Errata to "On the convergence of the decomposition method for support vector machines"abstractProspective authors are requested to submit new, unpublished manuscripts for inclusion in the upcoming event described in this call for papers. Chih-Jen Lin |
IEEE Trans. Neural Networks | 1 |
| 2002 | Errata to "A comparison of methods for multiclass support vector machines"abstractProspective authors are requested to submit new, unpublished manuscripts for inclusion in the upcoming event described in this call for papers. Chih-Jen Lin |
IEEE Trans. Neural Networks | 1 |
| 2002 | A formal analysis of stopping criteria of decomposition methods for support vector machinesabstractIn a previous paper, the author (2001) proved the convergence of a commonly used decomposition method for support vector machines (SVMs). However, there is no theoretical justification about its stopping criterion, which is based on the gap of the violation of the optimality condition. It is essential to have the gap asymptotically approach zero, so we are sure that existing implementations stop in a finite number of iterations after reaching a specified tolerance. Here, we prove this result and illustrate it by two extensions: /spl nu/-SVM and a multiclass SVM by Crammer and Singer (2001). A further result shows that, in final iterations of the decomposition method, only a particular set of variables are still being modified. This supports the use of the shrinking and caching techniques in some existing implementations. Finally, we prove the asymptotic convergence of a decomposition method for this multiclass SVM. Discussions on the difference between this convergence proof and the one in another paper by Lin are also included. Chih-Jen Lin |
IEEE Trans. Neural Networks | 1 |
| 2001 | Optical Coating Designs Using the Family Competition Evolutionary AlgorithmabstractA robust evolutionary approach, called the Family Competition Evolutionary Algorithm (FCEA), is described for the synthesis of optical thin-film designs. Based on family competition and adaptive rules, the proposed approach consists of global and local strategies by integrating decreasing mutations and self-adaptive mutations. The method is applied to three different optical coating designs with complex spectral quantities. Numerical results indicate that the proposed approach performs very robustly and is very competitive with other approaches. Jinn-Moon Yang, Jorng-Tzong Horng, Chih-Jen Lin, Cheng-Yan Kao |
Evol. Comput. | 3 |
| 2001 | Training nu-Support Vector Classifiers: Theory and AlgorithmsabstractThe nu-support vector machine (nu-SVM) for classification proposed by Schölkopf, Smola, Williamson, and Bartlett (2000) has the advantage of using a parameter nu on controlling the number of support vectors. In this article, we investigate the relation between nu-SVM and C-SVM in detail. We show that in general they are two different problems with the same optimal solution set. Hence, we may expect that many numerical aspects of solving them are similar. However, compared to regular C-SVM, the formulation of nu-SVM is more complicated, so up to now there have been no effective methods for solving large-scale nu-SVM. We propose a decomposition method for nu-SVM that is competitive with existing methods for C-SVM. We also discuss the behavior of nu-SVM by some numerical experiments. Chih-Chung Chang, Chih-Jen Lin |
Neural Comput. | 2 |
| 2001 | Formulations of Support Vector Machines: A Note from an Optimization Point of ViewabstractIn this article, we discuss issues about formulations of support vector machines (SVM) from an optimization point of view. First, SVMs map training data into a higher- (maybe infinite-) dimensional space. Currently primal and dual formulations of SVM are derived in the finite dimensional space and readily extend to the infinite-dimensional space. We rigorously discuss the primal-dual relation in the infinite-dimensional spaces. Second, SVM formulations contain penalty terms, which are different from unconstrained penalty functions in optimization. Traditionally unconstrained penalty functions approximate a constrained problem as the penalty parameter increases. We are interested in similar properties for SVM formulations. For two of the most popular SVM formulations, we show that one enjoys properties of exact penalty functions, but the other is only like traditional penalty functions, which converge when the penalty parameter goes to infinity. Chih-Jen Lin |
Neural Comput. | 1 |
| 2001 | On the convergence of the decomposition method for support vector machinesabstractThe decomposition method is currently one of the major methods for solving support vector machines (SVM). Its convergence properties have not been fully understood. The general asymptotic convergence was first proposed by Chang et al. However, their working set selection does not coincide with existing implementation. A later breakthrough by Keerthi and Gilbert (2000, 2002) proved the convergence finite termination for practical cases while the size of the working set is restricted to two. In this paper, we prove the asymptotic convergence of the algorithm used by the software SVM(light) and other later implementation. The size of the working set can be any even number. Extensions to other SVM formulations are also discussed. Chih-Jen Lin |
IEEE Trans. Neural Networks | 1 |
| 2000 | The analysis of decomposition methods for support vector machinesabstractThe support vector machine (SVM) is a new and promising technique for pattern recognition. It requires the solution of a large dense quadratic programming problem. Traditional optimization methods cannot be directly applied due to memory restrictions. Up to now, very few methods can handle the memory problem and an important one is the "decomposition method." However, there is no convergence proof so far. In this paper, we connect this method to projected gradient methods and provide theoretical proofs for a version of decomposition methods. An extension to bound-constrained formulation of SVM is also provided.We then show that this convergence proof is valid for general decomposition methods if their working set selection meets a simple requirement. Chih-Chung Chang, Chih-Jen Lin |
IEEE Trans. Neural Networks Learn. Syst. | 3 |
| 1998 | Efficient test-point selection for scan-based BISTabstractWe propose a test point selection algorithm for scan-based built-in self-test (BIST). Under a pseudorandom BIST scheme, the objectives are (1) achieving a high random pattern fault coverage, (2) reducing the computational complexity, and (3) minimizing the performance as well as the area overheads due to the insertion of test points. The proposed algorithm uses a hybrid approach to accurately estimate the profit of the global random testability of a test point candidate. The timing information is fully integrated into the algorithm to access the performance impact of a test point. In addition, a symbolic procedure is proposed to compute testability measures more efficiently for circuits with feedbacks so that the test point selection algorithm can be applied to partial-scan circuits. The experimental results show the proposed algorithm achieves higher fault coverages than previous approaches,with a significant reduction of computational complexity. By taking timing information into consideration, the performance degradation can he minimized with possibly more test points. Huan-Chih Tsai, Kwang-Ting Cheng, Chih-Jen Lin, Sudipta Bhawmik |
IEEE Trans. Very Large Scale Integr. Syst. | 3 |
| 1997 | A Hybrid Algorithm for Test Point Selection for Scan-Based BISTabstractWe propose a new algorithm for test point selection for scan-based BIST. The new algorithm combines the advantages of both explicit-testability-calculation and gradient techniques. The test point selection is guided bya cost function which is partially based on explicit testability recalculation and partially on gradients. With an event-driven mechanism, it can quickly identify a set of nodes whose testability need to be recalculated due to a test point, and then use gradients to estimate the impact of the rest of the circuit. In addition, by incorporating timing information into the cost function, timing penalty caused by test points can be easily avoided. We present the results to illustrate that high fault coverages for both area- and timing-driven test point insertions can be obtained with a small number of test points. The results also indicate a signi#cant reduction of computational complexity while the qualities are similar to the explicitly-testability-calculation method. 1 I... Huan-Chih Tsai, Kwang-Ting Cheng, Chih-Jen Lin, Sudipta Bhawmik |
DAC | 3 |
| 1995 | Timing-Driven Test Point Insertion for Full-Scan and Partial-Scan BISTabstractWe propose timing-driven test point insertion methods for a full-scan based BIST scheme and for a partial-scan based BIST scheme, where the global flip-flop cycles have been broken by the scan flip-flops. The objective is to minimize the performance as well as the area impact due to the insertion of test points while achieving a high fault coverage under the pseudo-random BIST scheme. The gradient-based method is used and extended to estimate the random-pattern testability improvement factors for the test point candidates of either full-scan based or partial-scan based BIST. We also propose a symbolic computation technique to compute testability for circuits under the partial-scan based BIST scheme. Experimental results show that the performance degradation of test point insertion could be unacceptably high if the cost function used for test point selection does not include the performance penalty. Using our timing-driven algorithm, zero performance degradation and a high fault coverage can always be achieved using a small number of test points. Kwang-Ting Cheng, Chih-Jen Lin |
ITC | 2 |
| 1995 | Integration of partial scan and built-in self-test
Chih-Jen Lin, Yervant Zorian, Sudipta Bhawmik |
J. Electron. Test. | 1 |
| 1993 | Built-In Current Sensor for IDDQ Test in CMOSabstractThis paper presents a current sensor circuit which can be built into a CMOS logic circuit to perform a self test for leakage current. The distinct features of the current sensor circuitry are described in detail. The circuit is verified by using the SPICE2 simulator.> Ching-Wen Hsue, Chih-Jen Lin |
ITC | 2 |
| 1993 | PSBIST: A Partial-Scan Based Built-In Self-Test SchemeabstractPartial-scan based built-in self-test (PSBIST) is a versatile design for testability (DFT) scheme, which employs pseudo-random BIST at all levels of test to achieve fault coverages greater than 98% on average, and supports deterministic partial scan at the IC level to achieve nearly 100% fault coverage. While PSBIST provides all the benefits of BIST, it incurs less area overhead and performance degradation than full scan. The area overhead is further reduced when the boundary scan cells are reconfigured for BIST usage.> Chih-Jen Lin, Yervant Zorian, Sudipta Bhawmik |
ITC | 1 |
| 1991 | Enhanced Controllability for IDDQ Test Sets Using Partial Scan
Tapan J. Chakraborty, Sudipta Bhawmik, Robert Bencivenga, Chih-Jen Lin |
DAC | 4 |