English

Learning to Weight for Text Classification

Machine Learning 2021-09-22 v1 Information Retrieval Machine Learning

Abstract

In information retrieval (IR) and related tasks, term weighting approaches typically consider the frequency of the term in the document and in the collection in order to compute a score reflecting the importance of the term for the document. In tasks characterized by the presence of training data (such as text classification) it seems logical that the term weighting function should take into account the distribution (as estimated from training data) of the term across the classes of interest. Although `supervised term weighting' approaches that use this intuition have been described before, they have failed to show consistent improvements. In this article we analyse the possible reasons for this failure, and call consolidated assumptions into question. Following this criticism we propose a novel supervised term weighting approach that, instead of relying on any predefined formula, learns a term weighting function optimised on the training set of interest; we dub this approach \emph{Learning to Weight} (LTW). The experiments that we run on several well-known benchmarks, and using different learning methods, show that our method outperforms previous term weighting approaches in text classification.

Keywords

Cite

@article{arxiv.1903.12090,
  title  = {Learning to Weight for Text Classification},
  author = {Alejandro Moreo Fernández and Andrea Esuli and Fabrizio Sebastiani},
  journal= {arXiv preprint arXiv:1903.12090},
  year   = {2021}
}

Comments

To appear in IEEE Transactions on Knowledge and Data Engineering

R2 v1 2026-06-23T08:22:21.650Z