Learning-based Classification Research Articles

Evaluation of a Voltametric E-Tongue Combined with Data Preprocessing for Fast and Effective Machine Learning-Based Classification of Tomato Purées by Cultivar.

The potential of a voltametric E-tongue coupled with a custom data pre-processing stage to improve the performance of machine learning techniques for rapid discrimination of tomato purées between cultivars of different economic value has been investigated. To this aim, a sensor array with screen-printed carbon electrodes modified with gold nanoparticles (GNP), copper nanoparticles (CNP) and bulk gold subsequently modified with poly(3,4-ethylenedioxythiophene) (PEDOT), was developed to acquire data to be transformed by a custom pre-processing pipeline and then processed by a set of commonly used classifiers. The GNP and CNP-modified electrodes, selected based on their sensitivity to soluble monosaccharides, demonstrated good ability in discriminating samples of different cultivars. Among the different data analysis methods tested, Linear Discriminant Analysis (LDA) proved to be particularly suitable, obtaining an average F1 score of 99.26%. The pre-processing stage was beneficial in reducing the number of input features, decreasing the computational cost, i.e., the number of computing operations to be performed, of the entire method and aiding future cost-efficient hardware implementation. These findings proved that coupling the multi-sensing platform featuring properly modified sensors with the custom pre-processing method developed and LDA provided an optimal tradeoff between analytical problem solving and reliable chemical information, as well as accuracy and computational complexity. These results can be preliminary to the design of hardware solutions that could be embedded into low-cost portable devices.

Machine learning-based classification of structured light modes under turbulence and eavesdropping effects

This paper considers the classification of multiplexed structured light modes, aiming to bolster communication reliability and data transfer rates, particularly in challenging scenarios marked by turbulence and potential eavesdropping. An experimental free-space optic (FSO) system is established to transmit 16 modes [8-ary Laguerre Gaussian (LG) and 8-ary superposition LG (Mux-LG) mode patterns] over a 3-m FSO channel, accounting for interception threats and turbulence effects. To the best of authors’ knowledge, this paper is the first to consider both factors concurrently. We propose four machine/deep learning algorithms—artificial neural network, support vector machine, 1D convolutional neural network, and 2D convolutional neural network—for classification purposes. By fusing the outputs of these methods, we achieve promising classification results exceeding 92%, 81%, and 69% in cases of weak, moderate, and strong turbulence, respectively. Structured light modes exhibit significant potential for a variety of real-world applications where reliable and high-capacity data transmission is crucial.

Practical measurements distinguishing physiological and pathological stereoelectroencephalography channels based on high-frequency oscillations in the human brain.

The present study aimed to identify various distinguishing features for use in the accurate classification of stereoelectroencephalography (SEEG) channels based on high-frequency oscillations (HFOs) inside and outside the epileptogenic zone (EZ). HFOs were detected in patients with focal epilepsy who underwent SEEG. Subsequently, HFOs within the seizure-onset and early spread zones were defined as pathological HFOs, whereas others were defined as physiological. Three features of HFOs were identified at the channel level, namely, morphological repetition, rhythmicity, and phase-amplitude coupling (PAC). A machine-learning (ML) classifier was then built to distinguish two HFO types at the channel level by application of the above-mentioned features, and the contributions were quantified. Further verification of the characteristics and classifier performance was performed in relation to various conscious states, imaging results, EZ location, and surgical outcomes. Thirty-five patients were included in this study, from whom 166 104 pathological HFOs in 255 channels and 53 374 physiological HFOs in 282 channels were entered into the analysis pipeline. The results revealed that the morphological repetitions of pathological HFOs were markedly higher than those of the physiological HFOs; this was also observed for rhythmicity and PAC. The classifier exhibited high accuracy in differentiating between the two forms of HFOs, as indicated by an area under the curve (AUC) of 0.89. Both PAC and rhythmicity contributed significantly to this distinction. The subgroup analyses supported these findings. The suggested HFO features can accurately distinguish between pathological and physiological channels substantially improving its usefulness in clinical localization. In this study, we computed three quantitative features associated with HFOs in each SEEG channel and then constructed a machine learning-based classifier for the classification of pathological and physiological channels. The classifier performed well in distinguishing the two channel types under different levels of consciousness as well as in terms of imaging results, EZ location, and patient surgical outcomes.

Intelligent Stroke Disease Prediction Model Using Deep Learning Approaches.

Stroke is a high morbidity and mortality disease that poses a serious threat to people's health. Early recognition of the various warning signs of stroke is necessary so that timely clinical intervention can help reduce the severity of stroke. Deep neural networks have powerful feature representation capabilities and can automatically learn discriminant features from large amounts of data. This paper uses a range of physiological characteristic parameters and collaborates with deep neural networks, such as the Wasserstein generative adversarial networks with gradient penalty and regression network, to construct a stroke prediction model. Firstly, to address the problem of imbalance between positive and negative samples in the stroke public data set, we performed positive sample data augmentation and utilized WGAN-GP to generate stroke data with high fidelity and used it for the training of the prediction network model. Then, the relationship between observable physiological characteristic parameters and the predicted risk of suffering a stroke was modeled as a nonlinear mapping transformation, and a stroke prediction model based on a deep regression network was designed. Finally, the proposed method is compared with commonly used machine learning-based classification algorithms such as decision tree, random forest, support vector machine, and artificial neural networks. The prediction results of the proposed method are optimal in the comprehensive measurement index F. Further ablation experiments also show that the designed prediction model has certain robustness and can effectively predict stroke diseases.

Comprehensive quantitative radiogenomic evaluation reveals novel radiomic subtypes with distinct immune pattern in glioma

BackgroundAccurate classification of gliomas is critical to the selection of immunotherapy, and MRI contains a large number of radiomic features that may suggest some prognostic relevant signals. We aim to predict new subtypes of gliomas using radiomic features and characterize their survival, immune, genomic profiles and drug response. MethodsWe initially obtained 341 images of 36 patients from the CPTAC dataset for the development of deep learning models. Further 1812 images of 111 patients from TCGA_GBM and 152 images of 53 patients from TCGA_LGG were collected for testing and validation. A deep learning method based on Mask R–CNN was developed to identify new subtypes of glioma patients and compared the survival status, immune infiltration patterns, genomic signatures, specific drugs, and predictive models of different subtypes. Results200 glioma patients (mean age, 33 years ± 19 [standard deviation]) were enrolled. The accuracy of the deep learning model for identifying tumor regions achieved 88.3 % (98/111) in the test set and 83 % (44/53) in the validation set. The sample was divided into two subtypes based on radiomic features showed different prognostic outcomes (hazard ratio, 2.70). According to the results of the immune infiltration analysis, the subtype with a poorer prognosis was defined as the immunosilencing radiomic (ISR) subtype (n = 43), and the other subtype was the immunoactivated radiomic (IAR) subtype (n = 53). Subtype-specific genomic signatures distinguished celllines into ISR celllines (n = 9) and control celllines (n = 13), and identified eight ISR-specific drugs, four of which were validated by the OCTAD database. Three machine learning-based classifiers showed that radiomic and genomic co-features better predicted the radiomic subtypes of gliomas. ConclusionsThese findings provide insights into how radiogenomic could identify specific subtypes that predict prognosis, immune and drug sensitivity in a non-invasive manner.

Machine learning and radiomics analysis by computed tomography in colorectal liver metastases patients for RAS mutational status prediction.

To assess the efficacy of machine learning and radiomics analysis by computed tomography (CT) in presurgical setting, to predict RAS mutational status in colorectal liver metastases. Patient selection in a retrospective study was carried out from January 2018 to May 2021 considering the following inclusion criteria: patients subjected to surgical resection for liver metastases; proven pathological liver metastases; patients subjected to enhanced CT examination in the presurgical setting with a good quality of images; and RAS assessment as standard reference. A total of 851 radiomics features were extracted using the PyRadiomics Python package from the Slicer 3D image computing platform after slice-by-slice segmentation on CT portal phase by two expert radiologists of each individual liver metastasis performed first independently by the individual reader and then in consensus. Balancing technique was performed, and inter- and intraclass correlation coefficients were calculated to assess the between-observer and within-observer reproducibility of features. Receiver operating characteristics (ROC) analysis with the calculation of area under the ROC curve (AUC), sensitivity (SENS), specificity (SPEC), positive predictive value (PPV), negative predictive value (NPV) and accuracy (ACC) were assessed for each parameter. Linear and non-logistic regression model (LRM and NLRM) and different machine learning-based classifiers were considered. Moreover, features selection was performed before and after a normalized procedure using two different methods (3-sigma and z-score). Seventy-seven liver metastases in 28 patients with a mean age of 60years (range 40-80years) were analyzed. The best predictors, at univariate analysis for both normalized procedures, were original_shape_Maximum2DDiameter and wavelet_HLL_glcm_InverseVariance that reached an accuracy of 80%, an AUC ≥ 0.75, a sensitivity ≥ 80% and a specificity ≥ 70% (p value < < 0.01). However, a multivariate analysis significantly increased the accuracy in RAS prediction when a linear regression model (LRM) was used. The best performance was obtained using a LRM combining linearly 12 robust features after a z-score normalization procedure: AUC of 0.953, accuracy 98%, sensitivity 96%, specificity of 100%, PPV 100% and NPV 96% (p value < < 0.01). No statistically significant increase was obtained considering the tested machine learning both without normalization and with normalization methods. Normalized approach in CT radiomics analysis allows to predict RAS mutational status in colorectal liver metastases patients.

Improved tomato leaf disease classification through adaptive ensemble models with exponential moving average fusion and enhanced weighted gradient optimization.

Tomato is one of the most popular and most important food crops consumed globally. The quality and quantity of yield by tomato plants are affected by the impact made by various kinds of diseases. Therefore, it is essential to identify these diseases early so that it is possible to reduce the occurrences and effect of the diseases on tomato plants to improve the overall crop yield and to support the farmers. In the past, many research works have been carried out by applying the machine learning techniques to segment and classify the tomato leaf images. However, the existing machine learning-based classifiers are not able to detect the new types of diseases more accurately. On the other hand, deep learning-based classifiers with the support of swarm intelligence-based optimization techniques are able to enhance the classification accuracy, leading to the more effective and accurate detection of leaf diseases. This research paper proposes a new method for the accurate classification of tomato leaf diseases by harnessing the power of an ensemble model in a sample dataset of tomato plants, containing images pertaining to nine different types of leaf diseases. This research introduces an ensemble model with an exponential moving average function with temporal constraints and an enhanced weighted gradient optimizer that is integrated into fine-tuned Visual Geometry Group-16 (VGG-16) and Neural Architecture Search Network (NASNet) mobile training methods for providing improved learning and classification accuracy. The dataset used for the research consists of 10,000 tomato leaf images categorized into nine classes for training and validating the model and an additional 1,000 images reserved for testing the model. The results have been analyzed thoroughly and benchmarked with existing performance metrics, thus proving that the proposed approach gives better performance in terms of accuracy, loss, precision, recall, receiver operating characteristic curve, and F1-score with values of 98.7%, 4%, 97.9%, 98.6%, 99.97%, and 98.7%, respectively.

Pine Wilt Disease Segmentation with Deep Metric Learning Species Classification for Early-Stage Disease and Potential False Positive Identification

Pine Wilt Disease poses a significant global threat to forests, necessitating swift detection methods. Conventional approaches are resource-intensive but utilizing deep learning on ortho-mapped images obtained from Unmanned Aerial Vehicles offers cost-effective and scalable solutions. This study presents a novel method for Pine Wilt Disease detection and classification using YOLOv8 for segmenting diseased areas, followed by cropping the diseased regions from the original image and applying Deep Metric Learning for classification. We trained a ResNet50 model using semi-hard triplet loss to obtain embeddings, and subsequently trained a Random Forest classifier tasked with identifying tree species and distinguishing false positives. Segmentation was favored over object detection due to its ability to provide pixel-level information, enabling the flexible extension of subsequent bounding boxes. Deep Metric Learning-based classification after segmentation was chosen for its effectiveness in handling visually similar images. The results indicate a mean Intersection over Union of 83.12% for segmentation, with classification accuracies of 98.7% and 90.7% on the validation and test sets, respectively.

Single-cell classification based on label-free high-resolution optical data of cell adhesion kinetics

Selecting and isolating various cell types is a critical procedure in many applications, including immune therapy, regenerative medicine, and cancer research. Usually, these selection processes involve some labeling or another invasive step potentially affecting cellular functionality or damaging the cell. In the current proof of principle study, we first introduce an optical biosensor-based method capable of classification between healthy and numerous cancerous cell types in a label-free setup. We present high classification accuracy based on the monitored single-cell adhesion kinetic signals. We developed a high-throughput data processing pipeline to build a benchmark database of ~ 4500 single-cell adhesion measurements of a normal preosteoblast (MC3T3-E1) and various cancer (HeLa, LCLC-103H, MDA-MB-231, MCF-7) cell types. Several datasets were used with different cell-type selections to test the performance of deep learning-based classification models, reaching above 70–80% depending on the classification task. Beyond testing these models, we aimed to draw interpretable biological insights from their results; thus, we applied a deep neural network visualization method (grad-CAM) to reveal the basis on which these complex models made their decisions. Our proof-of-concept work demonstrated the success of a deep neural network using merely label-free adhesion kinetic data to classify single mammalian cells into different cell types. We propose our method for label-free single-cell profiling and in vitro cancer research involving adhesion. The employed label-free measurement is noninvasive and does not affect cellular functionality. Therefore, it could also be adapted for applications where the selected cells need further processing, such as immune therapy and regenerative medicine.

Development of a machine learning model to classify mineral deposits using sphalerite chemistry and mineral assemblages

Sphalerite is a commonly occurring mineral in natural systems and a prominent indicator mineral used in resource exploration. Its chemistry and associated mineral assemblages are both controlled by the physicochemical conditions of the local environment, such as temperature and sulfur fugacity. Accordingly, the chemistry of sphalerite and the nature of the associated minerals provide valuable clues about the classification of mineral deposits and their environment of formation. Nevertheless, exclusive reliance on the trace-element chemistry of sphalerite for deposit classification has its limitations given the multitude of factors (For example, deposit type, temperature, pressure, and background concentrations of elements) that affect its chemistry. To address this challenge, we develop machine learning models using the SHapley Additive exPlanations (SHAP) method to assess the importance of sphalerite trace-element chemistry and mineral assemblage information to distinguishing between five petrogenetically distinct mineral deposit types. This contribution demonstrates that a composite model that incorporates both data types markedly improve the accuracy of deposit type discrimination. The composite model is composed of two sub-models, Random Forest (RF) and Extra Random Trees (ERT), which are specifically employed for processing trace-element data and mineral assemblage data, respectively, due to their superior performance in handling these two distinct types of datasets. To simplify the end-user experience, we provide an executable file of the machine learning-based classifier, allowing it to be readily applied as an exploration tool using simple inputs. In summary, this work substantially enhances the confidence of Pb-Zn deposit type classification using sphalerite and introduces an innovative perspective on the application of machine learning to resolving complex geological problems.

Detecting Coseismic Landslides in GEE Using Machine Learning Algorithms on Combined Optical and Radar Imagery

Landslides, resulting from disturbances in slope equilibrium, pose a significant threat to landscapes, infrastructure, and human life. Triggered by factors such as intense precipitation, seismic activities, or volcanic eruptions, these events can cause extensive damage and endanger nearby communities. A comprehensive understanding of landslide characteristics, including spatio-temporal patterns, dimensions, and morphology, is vital for effective landslide disaster management. Existing remote sensing approaches mostly use either optical or synthetic aperture radar sensors. Integrating information from both these types of sensors promises greater accuracy for identifying and locating landslides. This study proposes a novel approach, the ML-LaDeCORsat (Machine Learning-based coseismic Landslide Detection using Combined Optical and Radar Satellite Imagery), that integrates freely available Sentinel-1, Palsar-2, and Sentinel-2 imagery data in Google Earth Engine (GEE). The approach also integrates relevant spectral indices and suitable bands used in a machine learning-based classification of coseismic landslides. The approach includes a robust and reproducible training and validation strategy and allows one to choose between five classifiers (CART, Random Forest, GTB, SVM, and Naive Bayes). Using landslides from four different earthquake case studies, we demonstrate the superiority of our approach over existing solutions in coseismic landslide identification and localization, providing a GTB-based detection accuracy of 87–92%. ML-LaDeCORsat can be adapted to other landslide events (GEE script is provided). Transfer learning experiments proved that our model can be applied to other coseismic landslide events without the need for additional training data. Our novel approach therefore facilitates quick and reliable identification of coseismic landslides, highlighting its potential to contribute towards more effective disaster management.

Raman spectroscopy for esophageal tumor diagnosis and delineation using machine learning and the portable Raman spectrometer

Esophageal cancer is one of the leading causes of cancer-related deaths worldwide. The identification of residual tumor tissues in the surgical margin of esophageal cancer is essential for the treatment and prognosis of cancer patients. But the current diagnostic methods, either pathological frozen section or paraffin section examination, are laborious, time-consuming, and inconvenient. Raman spectroscopy is a label-free and non-invasive analytical technique that provides molecular information with high specificity. Here, we report the use of a portable Raman system and machine learning algorithms to achieve accurate diagnosis of esophageal tumor tissue in surgically resected specimens. We tested five machine learning-based classification methods, including k-Nearest Neighbors, Adaptive Boosting, Random Forest, Principal Component Analysis-Linear Discriminant Analysis, and Support Vector Machine (SVM). Among them, SVM shows the highest accuracy (88.61 %) in classifying the esophageal tumor and normal tissues. The portable Raman system demonstrates robust measurements with an acceptable focal plane shift of up to 3 mm, which enables large-area Raman mapping on resected tissues. Based on this, we finally achieve successful Raman visualization of tumor boundaries on surgical margin specimens, and the Raman measurement time is less than 5 min. This work provides a robust, convenient, accurate, and cost-effective tool for the diagnosis of esophageal cancer tumors, advancing toward Raman-based clinical intraoperative applications.

Deep learning-based classification of anti-personnel mines and sub-gram metal content in mineralized soil (DL-MMD)

De-mining operations are of critical importance for humanitarian efforts and safety in conflict-affected regions. In this paper, we address the challenge of enhancing the accuracy and efficiency of mine detection systems. We present an innovative Deep Learning architecture tailored for pulse induction-based Metallic Mine Detectors (MMD), so called DL-MMD. Our methodology leverages deep neural networks to distinguish amongst nine distinct materials with an exceptional validation accuracy of 93.5%. This high level of precision enables us not only to differentiate between anti-personnel mines, without metal plates but also to detect minuscule 0.2-g vertical paper pins in both mineralized soil and non-mineralized environments. Moreover, through comparative analysis, we demonstrate a substantial 3% and 7% improvement (approx.) in accuracy performance compared to the traditional K-Nearest Neighbors and Support Vector Machine classifiers, respectively. The fusion of deep neural networks with the pulse induction-based MMD not only presents a cost-effective solution but also significantly expedites decision-making processes in de-mining operations, ultimately contributing to improved safety and effectiveness in these critical endeavors.

Deep learning-based classification of the capillary ultrastructure in human skeletal muscles

Deep learning-based classification of the capillary ultrastructure in human skeletal muscles

Noninvasive Molecular Subtyping of Pediatric Low-Grade Glioma with Self-Supervised Transfer Learning.

Purpose To develop and externally test a scan-to-prediction deep learning pipeline for noninvasive, MRI-based BRAF mutational status classification for pediatric low-grade glioma. Materials and Methods This retrospective study included two pediatric low-grade glioma datasets with linked genomic and diagnostic T2-weighted MRI data of patients: Dana-Farber/Boston Children's Hospital (development dataset, n = 214 [113 (52.8%) male; 104 (48.6%) BRAF wild type, 60 (28.0%) BRAF fusion, and 50 (23.4%) BRAF V600E]) and the Children's Brain Tumor Network (external testing, n = 112 [55 (49.1%) male; 35 (31.2%) BRAF wild type, 60 (53.6%) BRAF fusion, and 17 (15.2%) BRAF V600E]). A deep learning pipeline was developed to classify BRAF mutational status (BRAF wild type vs BRAF fusion vs BRAF V600E) via a two-stage process: (a) three-dimensional tumor segmentation and extraction of axial tumor images and (b) section-wise, deep learning-based classification of mutational status. Knowledge-transfer and self-supervised approaches were investigated to prevent model overfitting, with a primary end point of the area under the receiver operating characteristic curve (AUC). To enhance model interpretability, a novel metric, center of mass distance, was developed to quantify the model attention around the tumor. Results A combination of transfer learning from a pretrained medical imaging-specific network and self-supervised label cross-training (TransferX) coupled with consensus logic yielded the highest classification performance with an AUC of 0.82 (95% CI: 0.72, 0.91), 0.87 (95% CI: 0.61, 0.97), and 0.85 (95% CI: 0.66, 0.95) for BRAF wild type, BRAF fusion, and BRAF V600E, respectively, on internal testing. On external testing, the pipeline yielded an AUC of 0.72 (95% CI: 0.64, 0.86), 0.78 (95% CI: 0.61, 0.89), and 0.72 (95% CI: 0.64, 0.88) for BRAF wild type, BRAF fusion, and BRAF V600E, respectively. Conclusion Transfer learning and self-supervised cross-training improved classification performance and generalizability for noninvasive pediatric low-grade glioma mutational status prediction in a limited data scenario. Keywords: Pediatrics, MRI, CNS, Brain/Brain Stem, Oncology, Feature Detection, Diagnosis, Supervised Learning, Transfer Learning, Convolutional Neural Network (CNN) Supplemental material is available for this article. © RSNA, 2024.

An efficient Fusion-Purification Network for Cervical pap-smear image classification

Background and Objectives:In cervical cell diagnostics, autonomous screening technology constitutes the foundation of automated diagnostic systems. Currently, numerous deep learning-based classification techniques have been successfully implemented in the analysis of cervical cell images, yielding favorable outcomes. Nevertheless, efficient discrimination of cervical cells continues to be challenging due to large intra-class and small inter-class variations. The key to dealing with this problem is to capture localized informative differences from cervical cell images and to represent discriminative features efficiently. Existing methods neglect the importance of global morphological information, resulting in inadequate feature representation capability. Methods:To address this limitation, we propose a novel cervical cell classification model that focuses on purified fusion information. Specifically, we first integrate the detailed texture information and morphological structure features, named cervical pathology information fusion. Second, in order to enhance the discrimination of cervical cell features and address the data redundancy and bias inherent after fusion, we design a cervical purification bottleneck module. This model strikes a balance between leveraging purified features and facilitating high-efficiency discrimination. Furthermore, we intend to unveil a more intricate cervical cell dataset: Cervical Cytopathology Image Dataset (CCID). Results:Extensive experiments on two real-world datasets show that our proposed model outperforms state-of-the-art cervical cell classification models. Conclusions:The results show that our method can well help pathologists to accurately evaluate cervical smears.

Deep Learning-Based Classification and Semantic Segmentation of Lung Tuberculosis Lesions in Chest X-ray Images.

We present a deep learning (DL) network-based approach for detecting and semantically segmenting two specific types of tuberculosis (TB) lesions in chest X-ray (CXR) images. In the proposed method, we use a basic U-Net model and its enhanced versions to detect, classify, and segment TB lesions in CXR images. The model architectures used in this study are U-Net, Attention U-Net, U-Net++, Attention U-Net++, and pyramid spatial pooling (PSP) Attention U-Net++, which are optimized and compared based on the test results of each model to find the best parameters. Finally, we use four ensemble approaches which combine the top five models to further improve lesion classification and segmentation results. In the training stage, we use data augmentation and preprocessing methods to increase the number and strength of lesion features in CXR images, respectively. Our dataset consists of 110 training, 14 validation, and 98 test images. The experimental results show that the proposed ensemble model achieves a maximum mean intersection-over-union (MIoU) of 0.70, a mean precision rate of 0.88, a mean recall rate of 0.75, a mean F1-score of 0.81, and an accuracy of 1.0, which are all better than those of only using a single-network model. The proposed method can be used by clinicians as a diagnostic tool assisting in the examination of TB lesions in CXR images.

Web-based System for Identifying Fraudulent Job Advertisements with Machine Learning

Web-based System for Identifying Fraudulent Job Advertisements with Machine Learning

Machine Learning-Based Classification of Mulberry Leaf Diseases

Machine Learning-Based Classification of Mulberry Leaf Diseases

Image-driven machine learning for automatic characterization of grain size and distribution in smart vanadium dioxide thin films

This study focuses on characterizing grain size and distribution in vanadium dioxide (VO2) thin films using Atomic Force Microscope (AFM) images. Traditional segmentation methods face challenges in accurately detecting grains, and this paper addresses these limitations by employing machine learning-based classification approaches, defining three classes: grains, joints or dividing lines, and the background. Various classification methods, including supervised, unsupervised, and thresholding-based approaches, were investigated, demonstrating the superiority of machine learning-based methods over traditional segmentation techniques. Deep Learning (DL) models, such as Convolutional Neural Networks (CNN), Deep Belief Networks (DBN), and Deep Boltzmann Machines (DBM), achieved the best results in accuracy and precision. The DBN algorithm, in particular, outperformed other methods with a 93.66% accuracy in grain detection. This could be attributed to its probabilistic modeling and effective handling of challenging pixels, making it a powerful tool for accurate grain size and distribution analysis.