Exploring Feature Selection and Cross-Validation for Improved Crop Yield Prediction
Machine learning models find significant importance in Geographic Information Systems relating to agriculture. UAV-derived orthomosaics, digital height maps, and vegetative indices provide valuable data for predicting crop yields across regions. These insights can support informed decision-making for farmers and stakeholders. While machine learning excels at analyzing large datasets, training and validating models on randomly distributed spatial data may introduce uncertainty and bias due to spatial autocorrelation. This research evaluates the effectiveness of random and spatial cross-validation (CV) techniques for crop yield prediction using machine learning models. The study utilizes dry yield data from hybrid maize collected over two years across two fields in Waseca, MN. Feature selection was employed to reduce model complexity when analyzing multiple vegetative indices. Results indicate that random CV outperforms spatial CV, and among the models tested, Extreme Gradient Boosting achieved the highest predictive performance.
Authors
- K.E. North
- Rahul Gomes (ORCID: https://orcid.org/0000-0002-5377-8196)
- Papia F. Rozario (ORCID: https://orcid.org/0000-0002-2066-9479)
- Grace McDonnell
- Gabriel Sipos
- G M Abraham
- Daniel Chvat
- Landon Dierkes
Institutions
- University of Wisconsin–Eau Claire (US)
- North Carolina State University (US)
- University of California, Los Angeles (US)
- Madison Area Technical College (US)
- El Camino College (US)
Publication Details
- Published
- 2025-05-29
- DOI
- https://doi.org/10.1109/eit64391.2025.11103692
- Primary Topic
- Smart Agriculture and AI
- Type
- article
- Field-Weighted Citation Impact
- 0.00
Funders
- National Science Foundation
- Office of Research and Sponsored Programs, Marquette University