Supervised Machine Learning: A Brief Primer.

Journal: Behavior therapy
Published Date:

Abstract

Machine learning is increasingly used in mental health research and has the potential to advance our understanding of how to characterize, predict, and treat mental disorders and associated adverse health outcomes (e.g., suicidal behavior). Machine learning offers new tools to overcome challenges for which traditional statistical methods are not well-suited. This paper provides an overview of machine learning with a specific focus on supervised learning (i.e., methods that are designed to predict or classify an outcome of interest). Several common supervised learning methods are described, along with applied examples from the published literature. We also provide an overview of supervised learning model building, validation, and performance evaluation. Finally, challenges in creating robust and generalizable machine learning algorithms are discussed.

Authors

  • Tammy Jiang
    Department of Epidemiology, Boston University School of Public Health, Boston, Massachusetts.
  • Jaimie L Gradus
    Department of Epidemiology, Boston University School of Public Health, Boston, Massachusetts.
  • Anthony J Rosellini
    Department of Health Care Policy, Harvard Medical School.