Skip to main navigation Skip to search Skip to main content

Overcoming the ordinal imbalanced data problem by combining data processing and stacked generalizations

Marine Desprez*, Kyle Zawada, Daniel Ramp

*Corresponding author for this work

Research output: Contribution to journalArticlepeer-review

19 Downloads (Pure)

Abstract

Ordinal imbalanced datasets are pervasive in real world applications but remain challenging to analyse as they require specific methods to account for the ordering information and imbalanced classes. Failure to account for both those characteristics can substantially impact the model predictive performance. However, existing methods tend to focus either on ordinality or imbalance, rather than addressing both simultaneously. The few approaches that do account for both characteristics are not always easy to implement for non-advanced analysts and simpler approaches are needed to facilitate appropriate data processing. Here, we developed a general approach using some of the most popular machine learning algorithms to ensure appropriate processing of ordinal imbalanced datasets and to optimize the predictions of all classes. After transforming the multi-class ordinal problem into a well-known binary problem, we implemented several different resampling methods in a decision-tree classifier. We then used a stacked generalization algorithm to combine the classifiers to improve model predictive performance. To test our approach, we used two ordinal imbalanced datasets on student performance and wine quality. Individual resampling techniques tended to improve the accuracy of minority classes, while simultaneously increasing the number of false positives in those classes. This resulted in a decrease, sometimes substantial, in accuracy of other classes. The stacking model offered a good compromise between improvement in accuracy of minority classes and mitigation of reduced accuracy in other classes. Our approach provided useful insights into modelling strategies that should be favoured for implementation in production that involve these common datasets, depending on the end-user interests.
Original languageEnglish
Article number100241
Pages (from-to)1-9
Number of pages9
JournalMachine Learning with Applications
Volume7
Early online date21 Dec 2021
DOIs
Publication statusPublished - 15 Mar 2022
Externally publishedYes

Bibliographical note

Copyright the Author(s) 2021. Version archived for private and non-commercial use with the permission of the author/s and according to publisher conditions. For further rights please contact the publisher.

Keywords

  • Stacked generalizations
  • Machine learning
  • Ordinal data
  • Imbalanced data
  • Random forests
  • Resampling methods
  • Rare events
  • Classification

Fingerprint

Dive into the research topics of 'Overcoming the ordinal imbalanced data problem by combining data processing and stacked generalizations'. Together they form a unique fingerprint.

Cite this