A Unified Framework for Joint Feature Selection and Hyperparameter Optimization in LSTM Networks using a Multi-Objective Modified Firefly Algorithm
The performance of Long Short-Term Memory (LSTM) networks in financial time series forecasting is heavily dependent on two interdependent factors: the selection of input features and the configuration of network hyperparameters. Existing methodologies treat these as isolated sequential problems, optimizing one while holding the other fixed, which guarantees a suboptimal convergence. This paper introduces a Multi-Objective Modified Firefly Algorithm (MOMFA) that executes joint feature selection and hyperparameter optimization as a single unified search. Candidate solutions are encoded as 30-dimensional vectors combining a discrete binary feature mask with continuous hyperparameter spaces. The standard Firefly Algorithm is modified via Lévy flight exploration and a sigmoid transfer function to bridge the continuous and discrete search spaces. Optimizing for both forecast accuracy (RMSE) and model complexity, the framework generates a Pareto-optimal set of network configurations.
Authors
- P Lalitha S Fernando
- H.V.A.S. Udayanga (ORCID: https://orcid.org/0009-0007-4052-0480)
Institutions
- University of Sri Jayewardenepura (LK)
Publication Details
- Journal
- Zenodo (CERN European Organization for Nuclear Research)
- Published
- 2026-05-12
- DOI
- https://doi.org/10.5281/zenodo.20132641
- Primary Topic
- Stock Market Forecasting Methods
- Type
- preprint