Estimation of minimal data sets sizes for machine learning predictions in digital mental health interventions

Publikation: Beiträge in ZeitschriftenZeitschriftenaufsätzeForschungbegutachtet

Authors

  • Kirsten Zantvoort
  • Barbara Nacke
  • Dennis Görlich
  • Silvan Hornstein
  • Corinna Jacobi
  • Burkhardt Funk

Artificial intelligence promises to revolutionize mental health care, but small dataset sizes and lack of robust methods raise concerns about result generalizability. To provide insights on minimal necessary data set sizes, we explore domain-specific learning curves for digital intervention dropout predictions based on 3654 users from a single study (ISRCTN13716228, 26/02/2016). Prediction performance is analyzed based on dataset size (N = 100–3654), feature groups (F = 2–129), and algorithm choice (from Naive Bayes to Neural Networks). The results substantiate the concern that small datasets (N ≤ 300) overestimate predictive power. For uninformative feature groups, in-sample prediction performance was negatively correlated with dataset size. Sophisticated models overfitted in small datasets but maximized holdout test results in larger datasets. While N = 500 mitigated overfitting, performance did not converge until N = 750–1500. Consequently, we propose minimum dataset sizes of N = 500–1000. As such, this study offers an empirical reference for researchers designing or interpreting AI studies on Digital Mental Health Intervention data.

OriginalspracheEnglisch
Aufsatznummer361
Zeitschriftnpj Digital Medicine
Jahrgang7
Ausgabenummer1
Anzahl der Seiten10
DOIs
PublikationsstatusErschienen - 12.2024

Bibliographische Notiz

Publisher Copyright:
© The Author(s) 2024.

DOI

Zuletzt angesehen

Forschende

  1. Martin Wendt

Publikationen

  1. Lifelong learning in practice at Leuphana University
  2. Trees in the desert
  3. Eye Movements During Mathematical Word Problem Solving-Global Measures and Individual Differences
  4. Higher Wages in Exporting Firms: Self-Selection, Export Effect, or Both?
  5. QALD-10 — The 10th Challenge on Question Answering over Linked Data
  6. The Lotka-Volterra Model for Competition Controlled by a Sliding Mode Approach
  7. Multitrophic diversity in a biodiverse forest is highly nonlinear across spatial scales
  8. Editorial overview
  9. Education and Communication as Prerequisites for and Components of Sustainable Development. Reflections for Policies, Conceptual Work, and Theory, Based on Previous Practises
  10. Does location really matter? An inter-colony comparison of seabirds breeding at varying distances from productive oceanographic features in the Bering Sea
  11. Lessons learned — The case of CROCUS
  12. Der "fachdidaktische Code" der Lebenswelt- und/oder (?) Situationsorientierung
  13. Modeling of microstructural pattern formation in crystal plasticity
  14. Enhancing the structural diversity between forest patches — A concept and real-world experiment to study biodiversity, multifunctionality and forest resilience across spatial scales
  15. Maintaining the Reputation of Reputation
  16. Exploring feedback and student characteristics relevant for personalizing feedback strategies
  17. Temperature changes using excimer laser irradiation in a cochlear model
  18. Knowledge Spaces of Globalization
  19. To Own or to Use?
  20. Applying FIDIC contracts in Jordan
  21. Public perceptions of CCS
  22. Logic
  23. Maschinenbelegungsplanung mit evolutionären Algorithmen