Transformer-based Short-term Drought Prediction with Feature Selection
Vol. 14, No. 4, pp. 282-288,
Apr. 2025
https://doi.org/10.3745/TKIPS.2025.14.4.282
PDF
Abstract
There have been many studies on short-term drought prediction. In the past, research mainly focused on numerical modeling. However,
due to the limitation of prediction accuracy, studies with artificial intelligence technology have recently gained traction. In the beginning,
machine learning models using a single algorithm and multilayer perceptron-based drought prediction research were conducted, and
then ensemble machine learning techniques and deep learning algorithms were introduced. Recently, transformer-based deep learning
algorithms, which have demonstrated outstanding performance, have garnered significant attention. In this study, longitudinal weather
observation data and standard precipitation data provided by the Meteorological Administration from 2015 to 2023 were preprocessed
and utilized. During the data pre-processing phrase, methods of processing missing values and outliers were implemented, new features
are generated from existing features, and feature extraction based on the wrapper method using a genetic algorithm and XGBoost is
performed. The key features were selected based on therir F1-score as the evaluation metic and were then used for machine learning
and deep learning experiments. The experimental results indicated that the transformer-based TFT deep learning algorithm, trained with
the selected features, achieved the highest F1-score of 0.88.
Statistics
Cite this article
[IEEE Style]
S. J. Hyun, "Transformer-based Short-term Drought Prediction with Feature Selection," The Transactions of the Korea Information Processing Society, vol. 14, no. 4, pp. 282-288, 2025. DOI: https://doi.org/10.3745/TKIPS.2025.14.4.282.
[ACM Style]
Seo Jae Hyun. 2025. Transformer-based Short-term Drought Prediction with Feature Selection. The Transactions of the Korea Information Processing Society, 14, 4, (2025), 282-288. DOI: https://doi.org/10.3745/TKIPS.2025.14.4.282.