English

The Application of Active Query K-Means in Text Classification

Computation and Language 2021-07-19 v1 Machine Learning Statistics Theory Statistics Theory

Abstract

Active learning is a state-of-art machine learning approach to deal with an abundance of unlabeled data. In the field of Natural Language Processing, typically it is costly and time-consuming to have all the data annotated. This inefficiency inspires out our application of active learning in text classification. Traditional unsupervised k-means clustering is first modified into a semi-supervised version in this research. Then, a novel attempt is applied to further extend the algorithm into active learning scenario with Penalized Min-Max-selection, so as to make limited queries that yield more stable initial centroids. This method utilizes both the interactive query results from users and the underlying distance representation. After tested on a Chinese news dataset, it shows a consistent increase in accuracy while lowering the cost in training.

Keywords

Cite

@article{arxiv.2107.07682,
  title  = {The Application of Active Query K-Means in Text Classification},
  author = {Yukun Jiang},
  journal= {arXiv preprint arXiv:2107.07682},
  year   = {2021}
}

Comments

6 pages, 3 algorithms, 4 tables, 8 figures For source code and questions, please email Yukun Jiang at jy2363@nyu.edu Reply would follow shortly

R2 v1 2026-06-24T04:15:01.588Z