Lähtökohdat Mielenterveysongelmat ovat merkittävä haaste työterveydelle. Analysoimme ­yleisluontoisen työterveyskyselyn vastausten ja mielenterveysdiagnoosien välistä yhteyttä ­koneoppimiseen perustuvalla luokittelulla.

Menetelmät Aineistona olivat 11 828 työterveyshuollon asiakkaan työterveyskyselyt sekä ­mielenterveysdiagnoosit. Ohjatun koneoppimisen XGBoost-luokittelijalla pyrittiin ennustamaan kyselyistä, saako henkilö diagnoosin kahden vuoden seurantajakson aikana.

Tulokset Sekä kaikkiin kyselyn kysymyksiin että seitsemään tärkeimpään kysymykseen perustuvat mallit ennustivat diagnooseja selvästi paremmin kuin ikään ja sukupuoleen perustuva triviaalimalli tai satunnaisluokittelija. Tärkeimmät diagnoosia ennustavat kysymykset liittyivät stressiin, ­surumielisyyteen ja väsymykseen.

Päätelmät Koneoppimisella onnistuttiin tunnistamaan mielenterveysongelmia ennustavia tekijöitä. Ongelman kannalta tärkeimmät kysymykset voidaan seuloa automaattisesti. Lähestymistapa saattaa osoittautua hyödylliseksi myös muissa työterveyden ja -turvallisuuden tutkimuksissa.