TraM : Enhancing User Sleep Prediction with Transformer-based Multivariate Time Series Modeling and Machine Learning Ensembles

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Kim, Jinjae, Ma, Minjeong, Choi, Eunjee, Cho, Keunhee, Lee, Chanwoo
Natura: Preprint
Pubblicazione: 2024
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866912073300049920
author Kim, Jinjae
Ma, Minjeong
Choi, Eunjee
Cho, Keunhee
Lee, Chanwoo
author_facet Kim, Jinjae
Ma, Minjeong
Choi, Eunjee
Cho, Keunhee
Lee, Chanwoo
contents This paper presents a novel approach that leverages Transformer-based multivariate time series model and Machine Learning Ensembles to predict the quality of human sleep, emotional states, and stress levels. A formula to calculate the labels was developed, and the various models were applied to user data. Time Series Transformer was used for labels where time series characteristics are crucial, while Machine Learning Ensembles were employed for labels requiring comprehensive daily activity statistics. Time Series Transformer excels in capturing the characteristics of time series through pre-training, while Machine Learning Ensembles select machine learning models that meet our categorization criteria. The proposed model, TraM, scored 6.10 out of 10 in experiments, demonstrating superior performance compared to other methodologies. The code and configuration for the TraM framework are available at: https://github.com/jin-jae/ETRI-Paper-Contest.
format Preprint
id arxiv_https___arxiv_org_abs_2410_11293
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle TraM : Enhancing User Sleep Prediction with Transformer-based Multivariate Time Series Modeling and Machine Learning Ensembles
Kim, Jinjae
Ma, Minjeong
Choi, Eunjee
Cho, Keunhee
Lee, Chanwoo
Machine Learning
Artificial Intelligence
This paper presents a novel approach that leverages Transformer-based multivariate time series model and Machine Learning Ensembles to predict the quality of human sleep, emotional states, and stress levels. A formula to calculate the labels was developed, and the various models were applied to user data. Time Series Transformer was used for labels where time series characteristics are crucial, while Machine Learning Ensembles were employed for labels requiring comprehensive daily activity statistics. Time Series Transformer excels in capturing the characteristics of time series through pre-training, while Machine Learning Ensembles select machine learning models that meet our categorization criteria. The proposed model, TraM, scored 6.10 out of 10 in experiments, demonstrating superior performance compared to other methodologies. The code and configuration for the TraM framework are available at: https://github.com/jin-jae/ETRI-Paper-Contest.
title TraM : Enhancing User Sleep Prediction with Transformer-based Multivariate Time Series Modeling and Machine Learning Ensembles
topic Machine Learning
Artificial Intelligence
url https://arxiv.org/abs/2410.11293