A Hybrid Framework Combining U-Net, Ant Colony Optimization, and CNN for Rice Leaf Disease Classification under Class Imbalance
DOI:
https://doi.org/10.29207/resti.v9i6.6910Keywords:
class imbalance, feature selection, hybrid sampling, image segmentation, precision agricultureAbstract
Accurate classification of rice plant diseases is essential for early intervention and precision agriculture. However, real-world datasets often suffer from complex backgrounds, high-dimensional features, and severe class imbalances, which compromise classification performance. This study proposes an integrated framework combining image segmentation using U-Net, feature selection via Ant Colony Optimization (ACO), hybrid sampling to handle class imbalance, and final classification using a Convolutional Neural Network (CNN). Segmentation isolates disease-affected areas, ACO optimizes feature subsets, and hybrid sampling balances class distribution using undersampling and SMOTE. The proposed method was tested on four rice leaf disease datasets—Brown Spot, Leaf Blast, Leaf Blight, and Leaf Scald—exhibiting significant class imbalance. Experimental results show that the proposed approach outperforms baseline models (SegNet, PspNet, and E-Net) across multiple metrics: Accuracy, IoU, Precision, and Recall. This indicates the framework’s robustness and potential for real-world deployment in precision agriculture. Future work will focus on model compression and real-time implementation in IoT systems.
Downloads
References
N. Hashim et al., “Smart Farming for Sustainable Rice Production: An Insight into Application, Challenge, and Future Prospect,” Rice Science, vol. 31, no. 1, pp. 47–61, Jan. 2024, doi: 10.1016/j.rsci.2023.08.004.
R. C. Joshi, R. Burget, and M. K. Dutta, “AgriDeep-net: An advanced deep feature fusion-based technique for enhanced fine-grain image analytics in precision agriculture,” Ecological Informatics, vol. 86, p. 103069, May 2025, doi: 10.1016/j.ecoinf.2025.103069.
Z. Salman, A. Muhammad, and D. Han, “Plant disease classification in the wild using vision transformers and mixture of experts,” Front. Plant Sci., vol. 16, June 2025, doi: 10.3389/fpls.2025.1522985.
W. Albattah and R. U. Khan, “Impact of imbalanced features on large datasets,” Front. Big Data, vol. 8, Mar. 2025, doi: 10.3389/fdata.2025.1455442.
L. Wang, X. Gong, J. Zhang, and R. Chen, “Anti-noise face: A resilient model for face recognition with labeled noise data,” Signal Processing: Image Communication, vol. 134, p. 117269, May 2025, doi: 10.1016/j.image.2025.117269.
K. Salazar-Serna, S. A. Barona, I. C. García, L. Cadavid, and C. J. Franco, “Addressing overfitting in classification models for transport mode choice prediction: a practical application in the Aburrá Valley, Colombia,” Transportation Letters, vol. 17, no. 7, pp. 1213–1230, Aug. 2025, doi: 10.1080/19427867.2024.2422717.
C. Kolling, V. Araujo, A. Veloso, and S. R. Musse, “Mitigating bias in facial analysis systems by incorporating label diversity,” Computers & Graphics, vol. 116, pp. 173–184, Nov. 2023, doi: 10.1016/j.cag.2023.08.021.
U. Muñoz-Aseguinolaza, I. Fernandez-Iriondo, I. Rodríguez-Moreno, N. Aginako, and B. Sierra, “Convolutional neural network-based classification and monitoring models for lung cancer detection: 3D perspective approach,” Heliyon, vol. 9, no. 11, p. e21203, Nov. 2023, doi: 10.1016/j.heliyon.2023.e21203.
S. Madhu, Dr. K. N. Bhargavi, M. V. S. Ramprasad, S. Gautam, and S. Bhavana, “Accurate diabetic retinopathy segmentation and classification model using gated recurrent unit with residual attention network,” Biomedical Signal Processing and Control, vol. 102, p. 107348, Apr. 2025, doi: 10.1016/j.bspc.2024.107348.
H. Vaidya et al., “A class imbalance aware hybrid model for accurate rice variety classification,” International Journal of Cognitive Computing in Engineering, vol. 6, pp. 170–182, Dec. 2025, doi: 10.1016/j.ijcce.2024.12.004.
S. Zarbakhsh, F. Fakhrzad, D. Rajkovic, G. Niedbała, and M. Piekutowska, “Approaches and challenges in machine learning for monitoring agricultural products and predicting plant physiological responses to biotic and abiotic stresses,” Current Plant Biology, vol. 43, p. 100535, Sept. 2025, doi: 10.1016/j.cpb.2025.100535.
A. Razzaque and D. A. Badholia, “PCA based feature extraction and MPSO based feature selection for gene expression microarray medical data classification,” Measurement: Sensors, vol. 31, p. 100945, Feb. 2024, doi: 10.1016/j.measen.2023.100945.
A. V. Jonnalagadda and H. A. Hashim, “SegNet: A segmented deep learning based Convolutional Neural Network approach for drones wildfire detection,” Remote Sensing Applications: Society and Environment, vol. 34, p. 101181, Apr. 2024, doi: 10.1016/j.rsase.2024.101181.
C. G. Simhadri, H. K. Kondaveeti, V. K. Vatsavayi, A. Mitra, and P. Ananthachari, “Deep learning for rice leaf disease detection: A systematic literature review on emerging trends, methodologies and techniques,” Information Processing in Agriculture, vol. 12, no. 2, pp. 151–168, June 2025, doi: 10.1016/j.inpa.2024.04.006.
M. Schotte, J. Chaumel, M. N. Dean, and D. Baum, “Image analysis pipeline for segmentation of a biological porosity network, the lacuno-canalicular system in stingray tesserae,” MethodsX, vol. 7, p. 100905, Jan. 2020, doi: 10.1016/j.mex.2020.100905.
Y. Yu et al., “Techniques and Challenges of Image Segmentation: A Review,” Electronics, vol. 12, no. 5, Art. no. 5, Jan. 2023, doi: 10.3390/electronics12051199.
A. Alweshah, R. Barzamini, F. Hajati, S. S. S. Farahani, M. Arabian, and B. Sohani, “Temporal dependency modeling for improved medical image segmentation: The R-UNet perspective,” Franklin Open, vol. 9, p. 100182, Dec. 2024, doi: 10.1016/j.fraope.2024.100182.
C. Zhang, X. Deng, and S. H. Ling, “Next-Gen Medical Imaging: U-Net Evolution and the Rise of Transformers,” Sensors, vol. 24, no. 14, Art. no. 14, Jan. 2024, doi: 10.3390/s24144668.
J. Albarracin, F. Cano, E. Romero, and A. Cruz-Roa, “A comparative analysis between two convolutional networks architectures for semantic segmentation of histopathology breast cancer images,” in 2023 19th International Symposium on Medical Information Processing and Analysis (SIPAIM), Nov. 2023, pp. 1–5. doi: 10.1109/SIPAIM56729.2023.10373477.
G. Bala, H. Mondal, and A. Sethi, “Brain MRI Segmentation Using U-Net and SegNet: A Comparative Study Across Modalities with Robust Loss Functions,” presented at the International Joint Conference on Biomedical Engineering Systems and Technologies, SCITEPRESS, Feb. 2025, pp. 246–254. doi: 10.5220/0013302900003911.
C. Beeche et al., “Super U-Net: A modularized generalizable architecture,” Pattern Recognition, vol. 128, p. 108669, Aug. 2022, doi: 10.1016/j.patcog.2022.108669.
B. Mustapha, Y. Zhou, B. Nawel, S. Chunyan, and X. Zhitao, “Optimized attention U-Net for enhanced lung and area of infection segmentation in chest X-Rays and CT scans,” Journal of Radiation Research and Applied Sciences, vol. 18, no. 3, p. 101650, Sept. 2025, doi: 10.1016/j.jrras.2025.101650.
Y. Li, K. Li, and Q. Lu, “Applying segmentation and classification to improve performance of smoothing,” Digital Signal Processing, vol. 109, p. 102913, Feb. 2021, doi: 10.1016/j.dsp.2020.102913.
S. Zhao, M. Wang, S. Ma, and Q. Cui, “A feature selection method via relevant-redundant weight,” Expert Systems with Applications, vol. 207, p. 117923, Nov. 2022, doi: 10.1016/j.eswa.2022.117923.
L. Tello-Cifuentes, J. Marulanda, and P. Thomson, “Segmentation and feature extraction-based classification of pavement damages using hybrid computer vision and machine learning approaches,” Mach Learn, vol. 114, no. 8, p. 172, June 2025, doi: 10.1007/s10994-025-06803-3.
M. Hassan and N. Kaabouch, “Impact of Feature Selection Techniques on the Performance of Machine Learning Models for Depression Detection Using EEG Data,” Applied Sciences, vol. 14, no. 22, Art. no. 22, Jan. 2024, doi: 10.3390/app142210532.
E. Oluwasakin et al., “Minimization of high computational cost in data preprocessing and modeling using MPI4Py,” Machine Learning with Applications, vol. 13, p. 100483, Sept. 2023, doi: 10.1016/j.mlwa.2023.100483.
Q. Zhou, Q. Gao, Q. Wang, M. Yang, and X. Gao, “Sparse discriminant PCA based on contrastive learning and class-specificity distribution,” Neural Networks, vol. 167, pp. 775–786, Oct. 2023, doi: 10.1016/j.neunet.2023.08.061.
A. Hashemi, M. Joodaki, N. Z. Joodaki, and M. B. Dowlatshahi, “Ant colony optimization equipped with an ensemble of heuristics through multi-criteria decision making: A case study in ensemble feature selection,” Applied Soft Computing, vol. 124, p. 109046, July 2022, doi: 10.1016/j.asoc.2022.109046.
R. Devi Priya, R. Sivaraj, N. Anitha, and V. Devisurya, “Tri-staged feature selection in multi-class heterogeneous datasets using memetic algorithm and cuckoo search optimization,” Expert Systems with Applications, vol. 209, p. 118286, Dec. 2022, doi: 10.1016/j.eswa.2022.118286.
S. Kashef and H. Nezamabadi-pour, “An advanced ACO algorithm for feature subset selection,” Neurocomputing, vol. 147, pp. 271–279, Jan. 2015, doi: 10.1016/j.neucom.2014.06.067.
H. Huang, H.-B. Xie, J.-Y. Guo, and H.-J. Chen, “Ant colony optimization-based feature selection method for surface electromyography signals classification,” Comput Biol Med, vol. 42, no. 1, pp. 30–38, Jan. 2012, doi: 10.1016/j.compbiomed.2011.10.004.
I.-S. Oh and J.-S. Lee, “Ant Colony Optimization for Feature Selection in Pattern Recognition,” The Journal of the Korea Contents Association, vol. 10, no. 5, pp. 1–9, 2010, doi: 10.5392/JKCA.2010.10.5.001.
A. Z. Ye et al., “High-Dimensional Feature Selection Based on Improved Binary Ant Colony Optimization Combined with Hybrid Rice Optimization Algorithm,” International Journal of Intelligent Systems, vol. 2023, no. 1, p. 1444938, 2023, doi: 10.1155/2023/1444938.
B. Chung, “Addressing Data Imbalance in Plant Disease Recognition through Contrastive Learning,” in 2024 IEEE 3rd International Conference on AI in Cybersecurity (ICAIC), Feb. 2024, pp. 1–6. doi: 10.1109/ICAIC60265.2024.10433841.
Y. Liu, L. Zhu, L. Ding, H. Sui, and W. Shang, “A hybrid sampling method for highly imbalanced and overlapped data classification with complex distribution,” Information Sciences, vol. 661, p. 120117, Mar. 2024, doi: 10.1016/j.ins.2024.120117.
N. Nnamoko and I. Korkontzelos, “Efficient treatment of outliers and class imbalance for diabetes prediction,” Artificial Intelligence in Medicine, vol. 104, p. 101815, Apr. 2020, doi: 10.1016/j.artmed.2020.101815.
D. Yilmaz Eroglu and M. S. Pir, “Hybrid Oversampling and Undersampling Method (HOUM) via Safe-Level SMOTE and Support Vector Machine,” Applied Sciences, vol. 14, no. 22, Art. no. 22, Jan. 2024, doi: 10.3390/app142210438.
M. Alamri and M. Ykhlef, “Hybrid Undersampling and Oversampling for Handling Imbalanced Credit Card Data,” IEEE Access, vol. 12, pp. 14050–14060, 2024, doi: 10.1109/ACCESS.2024.3357091.
Y. Ye et al., “Image segmentation using improved U-Net model and convolutional block attention module based on cardiac magnetic resonance imaging,” Journal of Radiation Research and Applied Sciences, vol. 17, no. 1, p. 100816, Mar. 2024, doi: 10.1016/j.jrras.2023.100816.
X. Zhang, K. Xu, L. Zhang, R. Zhao, W. Wei, and Y. She, “Optimal channel dynamic selection for Constructing lightweight Data EEG-based emotion recognition,” Heliyon, vol. 10, no. 9, p. e30174, May 2024, doi: 10.1016/j.heliyon.2024.e30174.
L. Chen, J. Jiang, Y. Zhang, and H. Chen, “HSDP: A Hybrid Sampling Method for Imbalanced Big Data Based on Data Partition,” Complex., vol. 2021, Jan. 2021, doi: 10.1155/2021/6877284.
R. Roboflow, “Palm Leaves Disease Detection,” Roboflow. Accessed: June 14, 2025. [Online]. Available: https://universe.roboflow.com/firstproject-ofadf/palm-leaves-disease-detection
M. Desai, H. Mewada, I. M. Pires, and S. Roy, “Evaluating the Performance of the YOLO Object Detection Framework on COCO Dataset and Real-World Scenarios,” Procedia Computer Science, vol. 251, pp. 157–163, Jan. 2024, doi: 10.1016/j.procs.2024.11.096.
L. Yan et al., “PSP net-based automatic segmentation network model for prostate magnetic resonance imaging,” Computer Methods and Programs in Biomedicine, vol. 207, p. 106211, Aug. 2021, doi: 10.1016/j.cmpb.2021.106211.
D. Wicker, M. M. Rizki, and L. A. Tamburino, “E-Net: Evolutionary neural network synthesis,” Neurocomputing, vol. 42, no. 1, pp. 171–196, Jan. 2002, doi: 10.1016/S0925-2312(01)00599-9.
Downloads
Published
Issue
Section
License
Copyright in each article belongs to the author
- The author acknowledges that the RESTI Journal (System Engineering and Information Technology) is the first publisher to publish with a license Creative Commons Attribution 4.0 International License.
- Authors can enter writing separately, arrange the non-exclusive distribution of manuscripts that have been published in this journal into other versions (eg sent to the author's institutional repository, publication in a book, etc.), by acknowledging that the manuscript has been published for the first time in the RESTI (Rekayasa Sistem dan Teknologi Informasi) journal ;
