A maximal figure-of-merit (MFoM)-learning approach to robust classifier design for text categorization

  • Authors:
  • Sheng Gao;Wen Wu;Chin-Hui Lee;Tat-Seng Chua

  • Affiliations:
  • Institute for Infocomm Research, Singapore;Carnegie Mellon University, Pittsburgh, PA;Georgia Institute of Technology, Atlanta, GA;National University of Singapore, Singapore

  • Venue:
  • ACM Transactions on Information Systems (TOIS)
  • Year:
  • 2006

Quantified Score

Hi-index 0.00

Visualization

Abstract

We propose a maximal figure-of-merit (MFoM)-learning approach for robust classifier design, which directly optimizes performance metrics of interest for different target classifiers. The proposed approach, embedding the decision functions of classifiers and performance metrics into an overall training objective, learns the parameters of classifiers in a decision-feedback manner to effectively take into account both positive and negative training samples, thereby reducing the required size of positive training data. It has three desirable properties: (a) it is a performance metric, oriented learning; (b) the optimized metric is consistent in both training and evaluation sets; and (c) it is more robust and less sensitive to data variation, and can handle insufficient training data scenarios. We evaluate it on a text categorization task using the Reuters-21578 dataset. Training an F1-based binary tree classifier using MFoM, we observed significantly improved performance and enhanced robustness compared to the baseline and SVM, especially for categories with insufficient training samples. The generality for designing other metrics-based classifiers is also demonstrated by comparing precision, recall, and F1-based classifiers. The results clearly show consistency of performance between the training and evaluation stages for each classifier, and MFoM optimizes the chosen metric.