Abstract
This study aims to address the challenge of detecting wood surface defects with high accuracy in real-world manufacturing environments, which are affected by various conditions. The goal is to build a hybrid deep learning model that can operate effectively in real time, serving automated quality control in wood processing. The research team proposes a hybrid architecture that combines multiple models: You Only Look Once – version 10 and CenterNet for accurate and fast localization of defect areas, Graph Attention Network to exploit spatial relationships between features, and Multilayer Perceptron for the final classification stage. At the same time, the Battle Royale Optimization algorithm is applied to select the optimal feature. The model is trained and tested on a large-scale dataset of 20,275 high-resolution wood surface images, covering 10 different types of defects. Experiments show that the model achieves a classification accuracy of 92.7% and a processing speed of 40 fps, meeting the requirements of real-time industrial systems. The model also demonstrates good generalization ability when effectively detecting both obvious defects and subtle abnormalities. The proposed hybrid modeling framework not only improves the efficiency of wood surface defect detection but also provides an automated, robust, and highly scalable solution for quality control in smart manufacturing. This contributes to optimizing the use of raw materials, minimizing waste, and enhancing the competitiveness of the wood processing industry.
Introduction
The surface quality of wood products is a key determinant of customer satisfaction, market competitiveness, and product longevity in the wood processing and furniture manufacturing industries. Surface defects such as cracks, wormholes, knots, and discoloration frequently arise due to machining errors, environmental exposure, or handling during transportation. These anomalies significantly compromise the esthetic appeal and structural integrity of final products, often resulting in increased material waste and decreased product value. 1 Traditional inspection methods such as visual assessment, tactile inspection, and manual measurements are still widely practiced and regulated (TCVN 1757:1975, TCVN 8931:2013), but these approaches are labor-intensive, prone to inconsistency, and unsuitable for high-throughput environments. The demand for scalable, real-time, and accurate defect detection solutions has grown significantly in light of Industry 4.0 trends and smart manufacturing initiatives. 2
In recent years, automated inspection systems utilizing image processing, 3D scanning, and surface roughness measurement tools have been introduced to address objectivity and repeatability issues. Nonetheless, ensuring consistent real-time performance in dynamic factory environments often characterized by varying lighting, surface textures, and mechanical vibrations remains challenging. 3 Deep learning models, especially Convolutional Neural Networks (CNNs), have shown strong potential in surface defect detection. However, single-stage detectors like You Only Look Once (YOLO) and Single Shot MultiBox Detector (SSD) offer speed at the expense of detection robustness, while two-stage models such as Faster R-CNN and Mask R-CNN provide greater accuracy but require high computational power, limiting their real-time deployment. Furthermore, many models suffer from poor generalization across different wood species and defect types, and they often struggle to capture spatial and contextual relationships between defect features.4,5 Although some hybrid approaches using optimization algorithms (such as Particle Swarm Optimization (PSO), Battle Royale Optimization (BRO)) or graph-based models (such as Graph Attention Network (GAT)) have been explored, there remains a lack of integrated solutions that balance speed, accuracy, and adaptability in production-scale applications.
Recent studies have shown the efficacy of Artificial Intelligence (AI) in mechanical fault detection and health management in various sectors. For instance, machine learning has been effectively applied in robotic rotary vector reducers and strain wave gear systems under variable operational speeds. Deep learning-based models like attention-guided feature aggregation networks have proven useful in diagnosing servo motor bearing faults, while transfer learning has enabled robust fault detection in industrial robotics. These findings underscore the significance of intelligent, adaptable, and data-driven approaches in real-time industrial applications, including wood defect inspection.
The research addresses the following questions: (1) How to balance real-time performance with accuracy in defect detection? (2) How to generalize detection across varying defect types and wood surfaces?. To overcome these limitations, this study proposes a novel hybrid framework for wood surface defect detection that integrates multiple advanced components: YOLOv10 and CenterNet for object detection, a Multi-Attention Graph Neural Network (MAGNN) for spatial-contextual learning, a Multi-Layer Perceptron (MLP) for final classification, and the BRO algorithm for feature selection. Among many available metaheuristic algorithms, we selected the BRO algorithm due to its ability to maintain population diversity and avoid local optima through competitive survival-based search strategies. Compared to more commonly used methods like PSO or Genetic Algorithm (GA). The BRO algorithm offers a balance between convergence speed and global search accuracy, which is critical for selecting optimal features from complex wood surface datasets. The system is further enhanced by the camera to support real-time defect localization under variable lighting and surface conditions. Unlike prior work, the proposed model leverages the complementary strengths of these components to achieve high detection accuracy, fast inference speed, and improved robustness across diverse wood textures and defect types. This paper offers the following key contributions:
Novel hybrid framework
We propose the first integration of YOLOv10 and CenterNet for defect detection, GAT for spatial-context learning, MLP for classification, and BRO for feature optimization, a configuration not previously explored in wood defect detection.
Real-time industrial application
Achieves high accuracy (92.7%) with 40 Frames Per Second (FPS), validated on a dataset of 20,275 industrial images under real-world conditions, surpassing existing models like YOLOv8 and Mask R-CNN.
Enhanced generalization
Our model addresses the limitation of poor adaptability in existing models by using GAT + BRO to robustly learn spatial and contextual relationships across varying wood species and defect types.
Impactful for smart manufacturing
Designed for easy integration into Industry 4.0 production lines, reducing defect-related waste and increasing product quality with minimal computational overhead.
This framework is intended to serve as a foundation for intelligent, automated quality control systems that minimize human intervention, reduce material waste, and enhance overall production efficiency. The paper is structured as follows: Section II explores related works. Section III elaborates on the proposed method. Section IV discusses experimental outcomes. Section V offers conclusions and future research directions.
Related works
Traditional inspection methods
Traditional inspection of wood surfaces largely relies on manual visual observation and tactile evaluation. While standardized by national codes such as TCVN 1757:1975 and TCVN 8931:2013, these methods are prone to subjectivity and inconsistency, especially when detecting small or hidden defects such as internal cracks, wormholes, or discoloration.6,7 Studies have shown that such methods are inefficient in high-throughput environments and struggle with accurate classification in complex defect scenarios.8,9
Machine learning approaches
Machine Learning (ML) has improved defect detection in terms of speed and automation. Techniques such as Support Vector Machines (SVM), Random Forest (RF), K-Nearest Neighbors (KNN), and Decision Trees have achieved accuracies ranging from 85% to 94%, depending on dataset size and wood type.10,11 However, these models often fail to generalize well to new or complex data, especially with small or overlapping defects, and typically require large, labeled datasets for reliable performance.12,13
Deep learning approaches
Deep Learning (DL), particularly Convolutional Neural Networks (CNNs), has demonstrated superior accuracy (90%–96%) in defect classification, thanks to automatic feature extraction from raw image data.14,15 Models such as Residual Network (ResNet), YOLO, EfficientNet, and CenterNet have been widely applied. Nevertheless, DL methods face challenges such as high computational costs, the need for Graphics Processing Unit (GPU) resources, and limited effectiveness when only Red–Green–Blue (RGB) images are used, especially for detecting subtle depth-related defects. 16 Although classical CNN-based models like ResNet and VGGNet show strong performance, they often struggle with spatial generalization and computational efficiency in high-speed environments. 17 Our method improves upon these by integrating GAT for spatial attention and BRO for optimal feature selection, achieving better balance in speed and accuracy.
Hybrid models and optimization algorithms
Recent studies propose hybrid models combining detection and optimization techniques. 18 The integration of YOLO with optimization methods like PSO or ACO (Ant Colony Optimization) has yielded improved classification rates, particularly in specific wood types.19,20 However, these models are often not scalable due to complexity and long computation times. Some recent efforts incorporate Graph Neural Network (GNN), which offer advantages in capturing spatial relationships and graph-structured data. Liu et al. 21 applied a CNN-GAT hybrid, improving classification performance to 99.37%. Yet, few studies have explored multi-attention GNNs combined with optimization strategies like BRO in real-time environments, highlighting the research gap this study addresses.
Proposed method
This study proposes a real-time model for identifying and classifying surface defects in wood during the final inspection stage on industrial conveyor belts. The model consists of three main components: image rule recognition, image annotation, model training, and real-time defect detection. Surface images are collected using industrial cameras and undergo preprocessing steps such as resizing, cropping, and labeling. The processed images are then used to train a deep learning model. The proposed approach combines a GAT within a GNN framework with a MLP for classification. GAT captures complex relationships between image regions, while MLP performs effective classification. This hybrid architecture allows for accurate and robust detection of wood surface defects in dynamic and uncertain manufacturing environments. Experimental results demonstrate that the proposed model significantly improves recognition performance and enables real-time application (Figure 1).

Overview of the proposed method.
Set
In this study, each wood surface image is represented as a graph
Data input
The input dataset used in this study consists of 20,275 high-resolution wood surface images (2800 × 1024 pixels) collected under real industrial conditions using a JAI SW-4000TL-PMCL line-scan camera at 66.4 fps. 22 Each image is annotated with bounding box labels and semantic segmentation maps identifying 10 common wood defects, including live knots, dead knots, cracks, resin, and blue stains. On average, each image contains 2.2 defects, with some having up to 16. Preprocessing steps include grayscale conversion, noise reduction, image binarization, Sobel edge detection, and histogram-based filtering to remove empty conveyor images. Cropping is performed to eliminate unnecessary backgrounds and reduce computation time. The dataset is well-structured with both pixel-level and object-level annotations, enabling applications in object detection, classification, and segmentation. Its quality and diversity make it suitable for training deep learning models in automated wood inspection systems.
Preprocessing
The testing and processing of wood images with defects involves a series of operations to rectify the issues. The simulated image of wood defects is processed using a digital image card, dividing the image into small pixels for efficient computer processing. The processed wood image, a digital lattice of 512 × 512 pixels, can be transformed by a computer to determine the gray value and brightness of each pixel. The image digitization process involved collecting and sampling wood defect images, resulting in an N × N array, which was then used for further analysis (equation (1)).
Median filter processing
The basic principle of median filter processing involves determining the value of a point in a digital image by analyzing the middle value of the surrounding area.
Calculate the central value of the array (
The median filter, a one-dimensional sliding window, determines the specific length of a point by adjusting the gray value of the middle pixel within the window. Let the digital image’s gray values be expressed as pixels, forming a two-dimensional median filter with a window of A. Calculate the value (y) in the array of two dimensions when setting
The binary processing of a gray image in wood involves converting it into a specific threshold (value between 0 and 255), converting it into a binary image with only two gray levels (0 value: black and 255 value: white), allowing for clearer change law representation. Edge points in an image are differentiated using a characteristic that identifies the most violent gray changes, thereby confirming edge pixels in the image through second-order differential analysis. Image processing employs differential operations for derivative operations, enhancing the universality of the operation by detecting specific direct edges. Calculate the gradient value of the surface of the wood image (equation (4)).
With:
The image gradient’s gray change is primarily observed in the gradient direction, with common operators like Sobel, Robert, Prewitt, Log, and Sobel assisting in edge detection, resulting in significant gray change. 23
The wood image processing process consists of two main stages: preprocessing and edge detection. First, the wood image is divided into small pixels (usually a 512 × 512 matrix) for easy analysis. Next, a Median Filter is applied to remove noise, which helps smooth the image without losing important information. The image is then binarized by setting a fixed threshold to convert it into two gray levels (black – 0 and white – 255), which helps highlight the defect area from the background image. In the edge detection stage, the gradient value is first calculated, measuring the change in brightness intensity between pixels to identify the area with the strongest variation, usually the defect border. Then, Sobel Edge Detector edge detection filters are applied to identify clear boundaries of the defects on the wood surface. The final result helps to clearly distinguish cracks, wood eyes, borers, or color changes, effectively supporting the process of checking wood quality with artificial intelligence. Theorem-type environments (including propositions, lemmas, corollaries, and more) can be formatted as follows:
YOLOv10
YOLOv10 integrates features from the Task-Aligned Learning (TAL) model and introduces dual label assignments to eliminate the need for Non-Maximum Suppression (NMS), improving training efficiency. 24 The model addresses the limitations of traditional NMS by applying a dimension gluing method and dual-task learning. Dual labeling helps maintain accuracy and convergence while reducing prediction errors. The two-headed architecture supports efficient optimization across varied wood surface types, lowering training costs and time. Overall, this combined approach significantly enhances automated wood quality inspection performance. Calculate the value of the consistent matching metric (equation (5)) to evaluate the level of agreement between the predicted value and the actual value when performing the training feature.
With:
Calculate the value that affects the labeling of the surface of the wood image and the monitoring of the two information heads. Calculate the objective function realization value for the classification operation to ensure accuracy in predicting hits that match the original targets. Calculate the distance value between two branches in supervision using the 1-Wasserstein distance to achieve the classification goal.
The YOLOv10 model features a lightweight classification head, spatial-channel decoupled down-sampling, and rank-guided block design. It enhances detection accuracy through large-kernel convolution and partial self-attention (PSA). Input and kernel sizes are carefully selected to balance processing speed and detail capture, with flexibility to adjust based on model requirements like CenterNet. YOLOv10 is highly suitable for wood image recognition due to its real-time, accurate detection of multiple defect types, including small and hard-to-spot flaws. It clearly distinguishes defect boundaries, ensuring precise classification. The model is optimized for deployment on resource-constrained devices, reducing hardware costs. This makes it ideal for integration into automated wood inspection systems, improving production efficiency and product quality.
Center net model
CenterNet 25 is an efficient object detection model that replaces traditional RoI pooling by using triplets of key pointscenter, and corners for accurate object localization. It builds on CornerNet by introducing center key points to refine and filter bounding boxes, improving detection precision. Bounding boxes are generated from top-k corner predictions and validated using center key point scores and positions. A central region is defined for each box, and only those containing valid center points are retained, with scores averaged from the three key points. The architecture includes calculating central weights, measuring mapping deviation, verifying centroid alignment, and maintaining bounding boxes based on deviation limits. The model’s performance is sensitive to the size of the central region; smaller areas reduce recall, while larger ones decrease precision, highlighting the importance of a scale-aware central region. CenterNet is well-suited for detecting defects in wood surface images, especially in handling variations in size and position.
Center pooling is a technique that captures richer, more recognizable visual patterns in geometric centers. This method, known as Enriching Center, involves finding the maximum value in both horizontal and vertical directions to determine if a pixel is a center key point, improving the detection of these key points. Corner Net utilizes the principle of cascade corner pooling to efficiently manage objects with outside appearance features, enhancing their overall functionality. The principle of pooling involves determining maximum values on boundary directions, but corners are sensitive to edges, necessitating the extraction of features from central regions. Cascade corner pooling utilizes a cascade approach to determine the maximum value along a boundary, subsequently adding it to obtain visual patterns (equation (6)).
With:
The robust model is trained from scratch using a high-resolution input image, minimizing training loss through data augmentation strategies. Single-scale testing involves detecting bounding boxes in images with original resolutions. In multi-scale testing, images with varying resolutions are input. Bounding boxes are detected using heatmaps and redundant ones. CenterNet is highly effective in wood image processing due to its ability to accurately detect the center and shape of small or unclear defects such as cracks, knots, and bugs. Its precision in identifying small, complex objects makes it ideal for industrial wood quality inspection. When combined with YOLOv10, the system enhances both defect detection and wood surface feature extraction. YOLOv10 detects large, obvious defects, while CenterNet focuses on smaller or ambiguous ones, allowing for comprehensive analysis. Together, they extract features like size, shape, position, and texture variations, which are crucial for classifying defects and assessing wood quality. These features support automated decision-making in inspection, predict future issues such as rot or cracking, and optimize production processes. This combined model significantly improves efficiency and accuracy in wood monitoring systems.
Battle Royale Optimization Algorithm (BRO)
The Battle Royale Optimization Algorithm (BRO) 26 effectively enhances feature selection for wood images after extraction by YOLOv10 and CenterNet. It identifies the most relevant features while eliminating redundant ones, reducing overfitting, and improving model generalization. BRO boosts computational efficiency, lowers system load, and increases processing speed for real-time wood quality monitoring. It also highlights critical features like cracks, knots, and beetles, improving classification accuracy. Integrating BRO with detection models enhances overall recognition performance and optimizes the wood inspection process (Shown in Figure 2).

Apply BRO in the image feature selection.
BRO is a competition-based game where players must survive and be the last man standing. The game’s objective is to eliminate opponents to win. Players must survive in a smaller and smaller play area, and resurrect if injured (equation (7)).
With:
Calculate the first initial value (equation (8)).
With:
The fitness level of neighboring solutions is compared, with the winner having the highest fitness level, resulting in a change in the position of the damaged solutions. The damage level of a solution in the search area is determined by its threshold, with the most damaged solution resetting to zero. The Battle Royale game utilizes the BRO algorithm to reduce the lower limits of d-dimensional space, ensuring a smaller playing area for optimal gameplay. Calculate the first iteration value. Calculate the reduction value in the spatial region. Calculate the upper bound and lower bound zone values.
The optimal outcome in a d-dimensional space is determined by updating the boundaries and standard deviation with the current location being the optimal outcome, and the best player/soldier considered an elite (equation (9)).
With:
The selection process involves identifying the best candidate solution, known as the winner. The fitness function minimizes the optimal solutions, thereby facilitating the pursuit of optimal solutions. The pseudo-code of BRO and the hyperparameter are shown in Table 1.
Hyperparameter analysis.
Graph attention Convolutional Network (GAT)
To enhance wood image recognition, a five-layer GAT model is used to learn important features such as texture, color, and geometric patterns characteristic of wood. GAT captures spatial relationships between pixels through graph-based node interactions, assigning attention weights to focus on relevant visual cues. 27 To optimize feature selection, the BRO algorithm is applied to evaluate and refine initial features like wood grain, brightness, and contrast. BRO calculates the fitness of each feature using an objective function, replacing low-performing features and adjusting search boundaries through repositioning and randomization. Features are selected based on accuracy and contribution to classification performance. This iterative process ensures that only the most relevant features are retained for training. As a result, the GAT model achieves better accuracy and robustness in wood recognition and quality analysis tasks.
This research paper proposes a GNN model to predict the surface of wood images through two stages: image feature extraction and the surface of wood image prediction stage. The first stage module of the proposed model uses a five-layer GAT model containing functional properties
The five-layer GAT model uses a hidden size of 128, ReLU activation, a dropout rate of 0.3, and is trained with the Adam optimizer (learning rate = 0.001) for 500 epochs. Each layer aggregates information from neighboring nodes using learned attention weights to capture spatial patterns between defects.
Feature extractor
The graph convolutional network utilizes an information aggregation function and an update function to generate the surface of wood informative representations, utilizing the surface of wood structure graph for learning and updating. Due to the different importance of neighbor nodes, weighted aggregation can yield more effective representations for the surface of the wood in a five-layer GAT model, generating embedding representations for each atomic node without manual feature engineering.
Information aggregation and update
The feature extractor uses convolutional operations to aggregate information and update node features in each layer of the surface of a wood structure graph, capturing crucial features of the surface of a wood functional group within each structure. To enhance readability and ensure consistent notation, we present a simplified description of the GAT mechanism applied in this study. Given a graph
With:
With:
The attention mechanism enables the model to selectively emphasize structurally important defect regions while suppressing irrelevant or noisy connections. After multiple GAT layers, a global pooling operation produces the graph-level embedding
Pooling of atomic feature vectors
Extracting the feature elements and initial point elements of the surface of the wood image forms the latent vector characteristic Z of each surface of the wood image of each multi-layer convolutional network node. At each latent layer, the information of neighboring elements
Feature aggregation for the surface of wood pairs
The feature vectors of the latent layers
The five-layer GAT model offers significant advantages in wood image classification by effectively capturing spatial relationships between pixels. Through its attention mechanism, GAT focuses on important features such as wood grain, color, and geometric patterns, improving detection accuracy. The model learns complex representations across layers, enhancing its ability to distinguish different types of wood. GAT also models correlations between features in graph space, enabling more robust classification. It minimizes the impact of noise, lighting variations, and viewing angles, maintaining high accuracy in challenging conditions. This makes it ideal for applications in material identification, wood quality analysis, and automated inspection in the wood manufacturing industry.
Multi-layer Perceptron (MLP) prediction
Multi-Layer Perceptron (MLP) prediction, 28 the GNN model performs the task of predicting the surface of wood images for the task of binary classification. The MLP model is found to be an excellent performance model in the classification task with five prediction classes. At the first 4 layers, the MLP model uses the ReLU model as the activation function, and at the last layer, the SoftMax function is used. The output mapping function in the range of 0–1 serves as the ability to detect latent classes of the surface of the wood image. The binary cross-entropy loss function (equation (13)) is used in the process of training the GNN network of the surface of wood image classification to continuously perform optimization for the research model.
With:
Combining an MLP method with a five-layer GAT enhances wood image classification by leveraging GAT’s strength in learning spatial relationships and MLP’s ability to handle complex, non-linear features. GAT captures subtle visual features like wood grain, color, and texture, while MLP refines and classifies them using fully connected layers with non-linear activation functions. This hybrid model improves accuracy and flexibility, especially in complex imaging conditions. In a production line application, a four-step process is proposed. Step 1 involves preprocessing and labeling wood images using normalization and data augmentation. Step 2 uses YOLOv10 to detect defects and CenterNet to determine object centers for better localization. Step 3 applies GAT to model feature relationships extracted from detection results, enhancing feature learning. Step 4 uses an MLP to classify wood types and conditions based on GAT-refined features. The system is expected to achieve 90%–95% classification accuracy, process 30–50 frames per second, and reduce manufacturing errors by 20%–30%. It also supports automation and continuous improvement using production data. In summary, the GAT module enhances defect classification by modeling spatial-contextual dependencies between defect regions rather than treating each region independently. By standardizing notation and clarifying the role of each component, the revised formulation aims to improve methodological transparency and accessibility without altering the underlying model design.
Results and experiment analysis
Experiment setup
Computer configuration parameters and hyperparameters are set during the process of conducting network training and testing the network of the Neural network model to verify the effectiveness of the research model. In this study, the hyperparameter settings were set to optimize the performance of the model on the wood surface dataset. The system uses an Intel Core i7-7600U processor, 32 GB RAM, and an NVIDIA GeForce RTX 3060 GPU (12 GB), ensuring powerful processing capabilities for the model training process. The programming environment was set on Ubuntu 20.04, with Python 3.7, CUDA 11.3, and TensorFlow 2.9, which helps to maximize the performance of the GPU during the computation process. To train the model effectively, the main parameters set includes batch size = 128, learning rate = 0.01, and number of epochs = 1000, allowing the model to learn well on large datasets while maintaining a stable convergence speed. The model uses an input image size of 640 × 640 pixels, ensuring the ability to detect detailed defects in each image. In addition, the Adam optimization algorithm is applied to effectively adjust the weights, helping the model achieve high accuracy during training. With this configuration, the system can process data quickly, optimize model performance, and improve the ability to detect wood surface defects in real-time
Datasets
The dataset used in this research is provided by Kodytek et al. 22 and comprises 20,275 high-resolution wood surface images (2800 × 1024 pixels) acquired in a real industrial sawmill environment. The acquisition system uses a JAI SW-4000TL-PMCL industrial line-scan camera operating at 66.4 frames per second, combined with a 3.5 million lux white-spectrum light source to ensure image clarity under high-speed conveyor motion and vibration. Each image is annotated with bounding boxes in .txt format and semantic segmentation maps in .bmp format, with a unique HEX color assigned to each of the 10 defect types.
The dataset includes both object-level and pixel-level annotations for 10 common wood surface defects, such as live knots, dead knots, cracks, resin spots, marrow, blue stains, and others (Shown in Figure 3). On average, each image contains 2.2 defects, with a maximum of 16 in a single image. Preprocessing involves grayscale conversion, noise filtering, binarization, and edge detection, followed by histogram-based filtering to remove empty conveyor images. An automatic cropping algorithm is used to isolate relevant image areas and reduce computational overhead (Shown in Tables 2 and 3).

Sample annotated images from the dataset.
Technical specifications and defect distribution.
The dataset is provided with an annotation color specification.
This dataset supports various deep learning tasks, including object detection using YOLOv10, CenterNet, feature extraction using GAT, MAGNN, and classification using the MLP method. The detailed annotations and wide range of defect types make it ideal for training models that require spatial awareness and multi-class classification capability. The variability in defect shape, size, and texture ensures robust learning and generalization, which is crucial for industrial applications with diverse wood species and lighting conditions.
Analysis of image segmentation
To assess the effectiveness of segmentation, three primary indicators are utilized: Accuracy, Sensitivity (Recall), and the Dice Similarity Coefficient (DSC). Accuracy quantifies the proportion of accurately classified pixels, encompassing both correctly identified positive (TrP) and negative pixels (TrN), relative to the overall pixel count (equation (14)). Sensitivity, also referred to as recall, represents the fraction of correctly detected positive pixels (TrP) in comparison to the sum of actual positive pixels (TrP + FsN; equation (15)). Dice Similarity Coefficient (DSC) is a commonly employed metric in image segmentation that measures the overlap between the predicted segmentation and the actual region. It determines how closely the segmented output aligns with the ground truth (equation (16)).
To ensure the accuracy and reliability of segmentation on the wood surface dataset, a structured evaluation process is essential. This involves comparing the original image, a grayscale preprocessed version, and the final segmented result to visually assess quality. Key metrics such as precision, recall, and DSC are calculated using ground truth images, allowing for pixel-wise comparison and performance evaluation. Preprocessing techniques like grayscale conversion, blurring, and edge enhancement help isolate key features and improve input quality. Knot segmentation is then performed using methods such as Otsu thresholding or edge filtering to identify even irregular contours. Cluster classification further distinguishes wood regions based on geometry and color. Adjacent region processing ensures accurate boundary separation when defects are located closely. Evaluating metrics across samples, including the standard deviation of DSC and accuracy, helps assess the method’s consistency. Low standard deviation indicates high stability, while high variation may suggest inconsistent performance. This comprehensive approach ensures the segmentation algorithm is both precise and practically applicable for real-world wood surface inspection (Table 4).
Results of image segmentation analysis.
From the image segmentation analysis results, the author can conclude that the segmentation method achieves high performance, with all DSC, Precision, Sensitivity, and Accuracy indices above 96%, demonstrating good accuracy and stability on the wood surface dataset. In particular, blue stain achieves the highest performance (98.36% DSC, 98.47% Precision, 98.49% Sensitivity, 98.38% Accuracy), showing that the algorithm is capable of clearly identifying this type of defect. However, crack (96.06% DSC) and knot missing (96.18% DSC) have slightly lower results, possibly due to complex shape characteristics or difficulty in determining clear boundaries. The stability of the method is demonstrated by the consistent accuracy across a wide range of defect types, along with a good balance between Precision and Sensitivity, indicating that the algorithm not only correctly identifies the segmentation region but also minimizes false positives. Although the overall results are very good, there are still slight variations between defect types, especially for cracks and missing knots, suggesting that further refinement is needed to improve the recognition performance. Further testing of advanced segmentation techniques or parameter optimization could help improve the accuracy on more difficult-to-distinguish defect types. The method has great potential for segmenting wood surface images, but there is still room for improvement to achieve the best performance across all defect types.
Image classification
The training of GNN was evaluated using two cross-validation strategies: edge set partition, where edges are randomly divided into 80% training and 20% test edges, and the surface of the wood image partition, where the surface of the wood image is randomly divided. The edge partition strategy involves dividing the training set into edge set A and the test set B. In the surface of the wood image partition strategy, the surface of the wood image is considered as a new surface of the wood image, while interactions among training the surface of the wood image are used as training samples. The surface of the wood image was divided into training and test samples, with edges between them serving as training samples. The surface of the wood image partition strategy was used to evaluate the performance of the predictor when a new surface of the wood image appeared, with average results used for evaluation.
In this evaluation, the effectiveness of the models for object detection is assessed using metrics such as accuracy (equation (17)), precision (Pr; equation (18)), recall (Rec; equation (19)), and the F1 score (equation (20)). The F1 score, a weighted harmonic mean of precision and recall, is used to evaluate performance due to the varying amounts of data for each insulator. These metrics are particularly crucial for binary classification problems, where the objective is to categorize instances into one of two classes: true answer or false answer. The metrics are defined as follows.
This study aims to evaluate the performance of the wood surface image classification method on the research dataset, thereby determining its applicability in practice. The classification results are analyzed based on evaluation criteria such as F1-score, Precision, and Sensitivity, which help determine the accuracy of the method from good to poor. According to Table 5, the Knot missing image has the best result, followed by Resin, while the Overgrown image type has the worst performance, which helps to identify the types of wood surfaces that are easy and difficult to classify. In addition, the Knot missing and Blue stain types have similar structures to Live knots, resulting in insignificant statistical differences. Reinforces the conclusion that Knot missing and Resin provide the best analysis performance, demonstrating that circular image architectures give more accurate classification results than curved or straight images. More importantly, the proposed method reduces the false recognition rate and the miss rate while maintaining high accuracy in monitoring the analytical limit. This confirms the feasibility of the method in practical applications, especially in automatic wood surface quality monitoring systems where high accuracy and performance are required.
Result of classification for wood surface image dataset.
Table 6 presents the training results of the wood surface defect detection algorithm with the datasets as described. Analyzing the detection results, the YOLO-v10 model gives the best results compared to the other two models, the SSD model and the BRO model, respectively, in terms of real-time image analysis (Figures 4 –7) and confusion matrix (Figure 8). This proves that the YOLO-v10 model is highly effective in identifying and identifying image objects in complex environments such as wood product manufacturing environments. Due to the complexity of the production environment, the precision value of the three models is smaller than the recall value of all three models. To improve the efficiency of the above problem, the task of the GAT model in the GNN architecture is a big challenge, so the proposed model has 5 GAT layers to solve the above problem. Due to the complexity of the production environment and uncertain working conditions, misidentification of objects sometimes results, which increases the value of the false positive (FP) score. Gradually, the value of the false positive (FP) score increases, but does not affect the recall value, and has a significant impact on the precision value (equation (17)), equation (20).
Detection evaluation.

The loss function.

Precision values.

Recall values.

F1-score values.

Confusion matric.
The results of identifying defect locations on the wood surface when performing model analysis in a complex wood product manufacturing environment. Labeled locations represent identifiable locations that are completely different from non-defect locations on the wood surface and are easily recognizable with the naked eye as defects on the surface of a wood panel within the camera’s light range industry. This research article brings high efficiency in applying computer vision techniques in practical production environments, bringing high economic efficiency. From the performance analysis results of the models, the author can draw some important conclusions about the effectiveness of each method in the problem of wood surface image segmentation. The YOLOv10 + CenterNet+ BRO + GNN model achieved the highest accuracy (0.91), Precision (0.96), Recall (0.97), and F1-score (0.97), showing that this combined method provides superior performance compared to other models. Although the proposed model architecture consists of multiple components (YOLOv10, CenterNet, GAT, MLP, BRO), each component performs a specific function to overcome the limitations of simple models. Specifically, YOLOv10 quickly handles large errors, CenterNet increases the ability to detect small errors, GAT learns spatial relationships, BRO removes noise, while MLP performs final classification. The integration of these components aims to achieve a balance between high accuracy, generalization ability, and real-time speed, something that simple models such as SVM, ResNet, or pure GCN cannot achieve simultaneously. The experimental results (Table 6) also demonstrate the superior performance of the proposed model.YOLOv8 (0.90 Accuracy, 0.92 Precision, 0.93 Recall, 0.92 F1-score) and Swin Transformer (0.89 Accuracy, 0.91 Precision, 0.92 Recall, 0.91 F1-score) also achieved impressive results, demonstrating the strong ability of advanced deep learning models in accurate image segmentation. Meanwhile, traditional models such as Random Forests (0.75 Accuracy, 0.73 Precision, 0.76 Recall, 0.74 F1-score) and SVM (0.77 Accuracy, 0.75 Precision, 0.78 Recall, 0.76 F1-score) have significantly lower performance, showing limitations when applied to highly complex problems such as wood surface segmentation. In addition, Mask R-CNN (0.87 Accuracy, 0.89 Precision, 0.90 Recall, 0.89 F1-score) and EfficientDetection (0.88 Accuracy, 0.90 Precision, 0.91 Recall, 0.90 F1-score) have good performance but are still lower than the YOLOv10 model combined with GNN. From this result, it can be seen that modern deep learning models, especially those combined with Graph Neural Networks (GNN), are capable of significantly improving the accuracy in recognizing and segmenting wood surface images. This confirms the effectiveness of GNN methods in optimizing segmentation results and opens up a new research direction in applying deep learning to monitor and inspect wood surface quality more automatically and accurately (Figure 9).

F1-score by wood defect subtype.
Crack defects tend to have irregular shapes and blurred edges, which confuse the segmentation step. Our GAT model’s attention mechanism improves robustness by focusing on texture continuity. Additionally, BRO enhances generalization by removing noisy features, which explains the observed increase in classification precision. Future evaluations will benchmark performance on various hardware platforms, including Jetson Nano, Raspberry Pi, and mid-range GPUs, to assess real-world deployment feasibility.
In addition to standard metrics (accuracy, precision, recall, F1-score), we evaluated the performance of the proposed model using ROC curves and AUC scores. As shown in Figure 10, the ROC curve for the proposed model achieves an AUC of 0.97, outperforming YOLOv8 (0.92) and EfficientDet (0.89). This indicates strong discriminative ability, especially in identifying subtle or ambiguous defects. ROC and AUC metrics further validate the robustness and practical utility of our framework in real-world applications (Figure 10).

ROC curve comparison.
YOLOv10 demonstrates improved performance over YOLOv8 and Mask R-CNN due to several architectural enhancements. First, its decoupled head and anchor-free mechanism separate classification and localization branches, reducing task interference and eliminating the need for anchor boxes, an advantage when detecting irregularly shaped defects such as cracks, sap stains, or surface indentations. Second, the incorporation of an edge-aware attention module enhances the model’s ability to focus on defect boundaries, which are often ambiguous and less effectively captured by YOLOv8 and Mask R-CNN. Lastly, YOLOv10 employs training refinements such as automatic learning of NMS thresholds and dynamic label assignment, making it better suited for imbalanced datasets like wood surface defects, where some defect classes are significantly underrepresented. Traditional models such as SVM and Random Forest are limited in their ability to learn spatial patterns from image data, as they primarily rely on handcrafted features or pre-extracted embeddings from CNNs. This limitation becomes significant when dealing with complex wood surface defects such as dead knots, edge cracks, or green mold, where geometric shapes and texture details are critical for accurate classification. Moreover, these models are highly sensitive to variations in lighting and complex backgrounds, which often occur in real-world production environments, leading to reduced stability and inconsistent performance.
Misclassifications frequently occur between certain defect types due to visual similarities. For instance, “crack” and “sap” defects are often confused because they both exhibit elongated shapes, indistinct boundaries, and similar light reflection characteristics. Likewise, “knot missing” and “dead knot” both appear as dark circular regions, making them difficult to distinguish, especially when the camera focus is uneven or under poor lighting conditions. Additionally, shadows, dust particles, or small bright spots on the wood surface can be mistakenly interpreted as cracks or knots, further contributing to false positives in detection. To evaluate the stability of the proposed model, the authors conducted five independent training runs using different random seeds. The results showed minimal variation, with an average accuracy of 91.03% (±0.34%) and an F1-score of 96.92% (±0.29%). These small standard deviations under 0.5%demonstrate the model’s consistency and reliability during training, confirming its robustness across multiple runs
Image feature selection evaluation
Feature selection is necessary to optimize the performance of the wood surface image classification model while minimizing the complexity of computation time and space. The analysis results in Table 7 show that applying this technique helps to comprehensively improve the evaluation indexes, including Accuracy, Precision, Sensitivity, Specificity, and F1-score. This proves that feature selection not only helps the learning model focus on important information but also eliminates unnecessary or noisy features, thereby improving the accuracy of wood surface classification. In addition, reducing the number of features also helps the system process faster, save resources, and improve the generalization ability of the model when applied to new data. Therefore, performing feature selection is an important step to optimize the model, which helps the wood surface classification system work more effectively in practice.
Analysis of image feature selection.
From the analysis results in Table 7, the author can draw important conclusions about the impact of feature selection on the performance of the wood surface image classification model. When applying feature selection, all evaluation indices are improved compared to when not using it, especially Accuracy (increased from 96.15 to 97.29), Precision (increased from 96.37 to 97.38), Specificity (increased from 97.39 to 97.89), and F1-score (increased from 97.28 to 97.73). This proves that feature selection not only helps the model focus on the most important features but also helps reduce noise, improving the accuracy and stability of the model. It is worth noting that the Sensitivity decreased slightly from 97.39 to 97.36, but this decrease was not significant and remained high, indicating that the feature selection did not reduce the ability to accurately identify positive samples. Furthermore, the Specificity increased significantly, indicating that the model was less affected by noisy samples or misclassification errors. From this result, the author can conclude that the application of feature selection improves the overall performance of the model, increasing the accuracy and correct classification ability while maintaining high sensitivity. This emphasizes the importance of feature optimization in image recognition systems, helping to minimize computational resources while still improving the performance of the wood surface classification system.
ANOVA statistical analysis
It is necessary to perform a statistical analysis of ANOVA to test whether there is a significant difference between the data groups, based on the evaluation metrics such as Accuracy, Precision, and Sensitivity. The results from Table 8 show that all the p-values are greater than 0.05, which means that there is no statistically significant difference between the data groups. This allows the author to conclude that the mean values of all the groups are considered to be similar, and any observed differences are not large enough to be considered statistically significant. Performing this analysis is important because it helps ensure that the changes in the data are not due to random factors but reflect the nature of the model or method being studied. If the ANOVA results show a statistically significant difference (p < 0.05), it may indicate that a certain group has a significantly better performance than the other groups. Conversely, when p > 0.05, as in this study, it means that the difference between groups is not large enough to conclude that one group is superior to the other. In short, performing ANOVA statistical analysis helps ensure that the results of the analysis are not affected by random factors, while also providing scientific evidence to determine whether the data groups are different. This helps improve the reliability of the study and helps to draw more accurate conclusions.
Result of analysis ANOVA.
From the ANOVA analysis results in Table 7, the author can conclude that there is no statistically significant difference between the data groups when considering the Accuracy, Precision, and Sensitivity indices, as all p-values are greater than 0.05 (Accuracy = 0.6271, Precision = 0.6926, Sensitivity = 0.6271). This suggests that the methods or models being compared have similar performance, and any observed differences may be due to random factors. Furthermore, the low F-value (Accuracy = 0.4023, Precision = 0.3017, Sensitivity = 0.4163) also confirms that the differences between the groups are not large enough to be statistically significant. In addition, the total variance within groups (Sum Square Within Groups (SSW)) is much higher than the variance between groups (Sum Square Between Groups (SSBG)), indicating that the variation within each group accounts for a larger proportion than the variation between groups. This suggests that the evaluated models perform similarly on this dataset, and the choice of which model may not significantly affect the final results. From these findings, the authors can conclude that if the goal is to find the best model, it may be necessary to extend the analysis by testing other criteria such as processing speed, generalization ability, or performance on real datasets. This result also raises the question of whether the current dataset and evaluation method are sensitive enough to distinguish the differences between models, and opens up new research directions to improve the accuracy of the analysis.
Computation time
In the process of processing wood surface images, the computation time of different models has significant differences in each processing step. Specifically, the YOLOv10 model, used to detect objects such as wood surfaces and defects, requires a relatively short computation time, only from 0.05 to 0.1 s. Next, CenterNet is used to determine the center position of objects, with a computation time of 0.03–0.08 s, which improves the accuracy of recognizing detailed features. The GAT model, which processes the relationship between features and learns the weights between features, requires more time, ranging from 0.15 to 0.3 s, but provides a powerful analysis of important features. Finally, the MLP method classifies the wood species and surface condition, requiring a calculation time of 0.02–0.05 s. The total calculation time for all these steps ranges from 0.25 to 0.5 s, which meets the fast processing speed requirements of real production conveyor systems, ensuring that the identification and classification process is performed efficiently and accurately. Comparison of wood image processing computation time between the proposed model (YOLOv10 + CenterNet + BRO + GAT + MLP) and other Machine Learning and Deep Learning models, adding information about image size (Image size: 512 × 512 pixels; Shown in Table 9).
Comparison of wood image processing computation time.
From the above studies, we can draw some important lessons related to the application of different models in wood image processing. The authors have applied many different methods to detect and classify wood image features, with each model having its advantages and limitations. YOLOv10 + CenterNet + BRO + GAT + MLP, the author’s proposed model, shows outstanding performance with fast computation time from 0.25 to 0.5 s, helping to improve the accuracy in object detection, and processing the relationship between features and image classification. Models such as SVM and Random Forest are also used for direct image classification, but require longer computation times, from 1 to 2 s for SVM and 0.5–1 s for Random Forest, and often only give good results when processing previously extracted features. Although K-Nearest Neighbors has a fairly long computation time (2–3 s), this model is still useful when dealing with smaller and simpler data. Deep Learning models such as CNN, ResNet, VGGNet, and Faster R-CNN are all capable of detecting objects and classifying wood images well, but require a computation time of 0.5–2.5 s, depending on the complexity of the model. From the above studies, it can be seen that the YOLOv10 + CenterNet + BRO + GAT + MLP model combining many advanced methods has helped to reduce the computation time, while optimizing the accuracy in processing wood images, demonstrating its superiority in real production environments.
Original contribution
This study provides an important step forward in the application of AI to wood surface quality control, with the YOLOv10 + CenterNet + BRO + GNN model achieving outstanding performance, reducing errors, increasing processing speed, and ensuring high accuracy. Experimental results show that the proposed model achieves Accuracy = 91%, Precision = 96%, Recall = 97%, and F1-score = 97%, outperforming traditional methods such as Mask R-CNN (Accuracy = 87%), EfficientDet (Accuracy = 88%) and YOLOv8 (Accuracy = 90%). In particular, the Battle Royale Optimization (BRO) algorithm improves Accuracy from 96.15% to 97.29%, Precision from 96.37% to 97.38%, and F1-score from 97.28% to 97.73%, demonstrating the effectiveness of feature selection in improving accuracy and reducing noise. Furthermore, the average processing time of the model is only 0.25–0.5 s, which is significantly faster than methods such as SVM (1–2 seconds) and Random Forest (1.5–2.5 s), showing the potential for practical implementation in automated quality control systems. When classifying wood surface defects, the best results belonged to Blue Stain with DSC = 98.36%, Precision = 98.47%, Sensitivity = 98.49%, and Accuracy = 98.38%, while Crack (DSC = 96.06%) and Knot Missing (DSC = 96.18%) had lower accuracy, indicating that further optimization is needed to detect defects with complex shapes. In addition, ANOVA analysis confirmed that the model had high stability, with all p-values >0.05 (Accuracy = 0.6271, Precision = 0.6926, Sensitivity = 0.6271), indicating uniformity in performance across different defect types. Thanks to the ability to reduce the defect rate to less than 2%, this model helps optimize the production process, save costs, and improve product quality, and can be integrated into smart production lines, combined with LiDAR sensors to detect small defects at low depth. With the results achieved, this research opens up a potential development direction in building automatic wood quality monitoring systems, helping the wood industry achieve higher efficiency in product control (Shown in Figure 11).

Performance distribution across the model.
To provide a visual summary of the model’s performance across multiple evaluation metrics, a boxplot was generated for accuracy, precision, recall, and F1-score. As shown, the F1-score and recall metrics demonstrate both high central values and low variability across models, indicating consistent model capability in detecting true positives and handling imbalanced data. Precision, however, exhibits greater variance, suggesting some models may generate more false positives, particularly those without advanced attention or optimization components. Notably, the proposed model (YOLOv10 + CenterNet + BRO + GNN) maintains high scores across all four metrics and shows minimal deviation, reinforcing its robustness and stability. This visualization highlights the performance gap between traditional models (SVM, Random Forest, and more) and modern deep learning-based architectures. The compact distribution of scores for models like EfficientDetection and YOLOv8 also reflects their maturity, while the broader spread in older or less specialized models confirms the need for multi-stage optimization in wood defect classification tasks (Shown in Figure 10).
Limitations of the proposed framework
Although the proposed model has many advantages in detecting and classifying wood surface defects, there are still some limitations that need to be considered to improve the performance in practical applications. One of the main disadvantages is that the computation time varies significantly at each processing step, which may affect the applicability in real production environments. In addition, the MAGNN model uses various attention mechanisms such as attention at the node, edge, and meta-path levels, which significantly increases the computational complexity, especially when applied to large-scale graphs. In addition, due to the characteristics of the attention mechanisms, the MAGNN model may be sensitive to noise or missing data, which affects the performance when applied to real-world datasets of uneven quality. Moreover, when the model is trained on small datasets or a limited number of samples, it is prone to overfitting, resulting in poor generalization ability. In addition, due to the MAGNN’s multiple network layers and complex aggregation mechanisms, the model requires large computational resources, including memory and training time, which may be difficult to deploy in large-scale production systems. To overcome these limitations, some future research directions can be considered, such as optimizing the aggregation matrix by calculating iteratively instead of directly or applying pruning techniques to reduce unimportant meta-paths to reduce the computational load. In addition, using regularization methods such as Dropout or L2 regularization can help the model learn better on noisy data. Future research can also apply transfer learning to leverage the weights that have been trained on large datasets, helping to improve the accuracy of the model on smaller datasets. While the proposed model achieves high accuracy and real-time performance, it faces challenges when detecting rare or subtle defects under extreme lighting conditions. Additionally, real-time deployment requires high computational resources, which may limit practical application in low-resource environments. Expanding the dataset and optimizing model efficiency are key directions for future research.
In addition to empirical runtime evaluation, we provide a theoretical computational complexity analysis of the proposed framework. The overall computational cost consists of three main components: detection, graph construction, and graph-based classification. For the detection stage, YOLOv10 operates with a complexity of approximately
Ablation study
The results in Table 10 show that the Full Model (YOLO + CenterNet + BRO + GAT + MLP) achieved the highest performance with Accuracy = 0.927, F1-score = 0.970, Recall = 0.970, and Precision = 0.960. When each component is removed, performance decreases significantly; specifically, without CenterNet, Recall = 0.870 and F1-score = 0.890 are achieved, while without GAT, the F1-score drops to 0.910. Comparing the two detectors separately, YOLOv10-only achieved Accuracy = 0.905 and Precision = 0.930, but Recall was only 0.910, while CenterNet-only had a higher Recall (0.930) but lower Precision (0.870). The combined YOLOv10 + CenterNet configuration achieved an Accuracy of 0.918 and an F1-score of 0.945, demonstrating that the dual-detector strategy improves overall performance compared to each model individually (Shown in Figure 12).
Ablation and dual-detector comparative results.

Ablation study.
Discussion
The experimental results confirm that the proposed hybrid framework integrating YOLOv10, CenterNet, GAT, BRO, and an MLP method effectively addresses the challenges of wood surface defect detection in real-time industrial environments. The system achieves superior performance across multiple evaluation metrics, including accuracy, precision, recall, and F1-score, when compared to traditional machine learning models and baseline deep learning approaches (Shown in Figure 13).

Performance comparison of wood defect detection models.
The combination of YOLOv10 and CenterNet provides a robust detection mechanism. Specifically, YOLOv10 demonstrates high efficiency in identifying prominent defects due to its fast and accurate bounding box regression, while CenterNet complements this by improving the localization of smaller or ambiguous defects through keypoint-based center estimation. The integration of BRO further enhances the model by performing effective feature selection, thereby reducing redundant inputs and optimizing the learning efficiency of the GNN module. The GAT model, in turn, captures complex spatial relationships between defect features, contributing significantly to improved classification outcomes. Despite these strengths, several limitations are observed. First, the model exhibits slightly reduced performance for irregular or low-contrast defects such as cracks and missing knots, which is attributed to their indistinct boundaries and complex geometric structures. Second, although the system maintains a real-time processing speed (0.25–0.5 s per image), the computational complexity introduced by BRO and GAT layers imposes constraints on lightweight or embedded deployment, particularly on systems without dedicated GPUs. Additionally, the current model is trained on two-dimensional image data, limiting its ability to detect depth-sensitive defects that might benefit from 3D imaging modalities.
To address these limitations, several improvements are proposed. Incorporating 3D imaging techniques such as stereo vision or LiDAR may enhance defect detection by introducing depth-awareness, particularly beneficial for subsurface anomalies. Moreover, integrating Internet of Things (IoT) devices can facilitate continuous, real-time defect monitoring and allow adaptive feedback mechanisms from production lines. Furthermore, applying model compression techniques such as pruning, quantization, or knowledge distillation can significantly reduce computational requirements, making the system more viable for deployment on resource-constrained embedded devices. The practical feasibility of deploying the proposed model in real-world industrial scenarios is also promising. The modular architecture allows for individual model components to be exported in optimized formats (ONNX, TensorRT, and more), supporting seamless integration with embedded AI platforms such as NVIDIA Jetson Nano, Xavier NX, and Google Coral. These edge platforms offer GPU-accelerated inference, enabling real-time classification with minimal latency and without reliance on cloud-based computation an advantage in latency-sensitive or bandwidth-limited environments. As outlined in Table 11, each hardware option offers trade-offs between power, throughput, and integration flexibility, enabling the model to be tailored to specific use cases and budget constraints. The proposed system not only demonstrates strong performance in experimental evaluation but also presents a viable path toward deployment in automated industrial quality control pipelines. Future research should focus on adaptive learning mechanisms, cross-domain generalization, and deeper integration with smart manufacturing infrastructures aligned with Industry 4.0 standards.
Embedded AI hardware options.
Conclusions
In this study, we proposed a novel hybrid deep learning framework combining YOLOv10, CenterNet, BRO, GAT, and MLP for automated wood surface defect detection. Evaluated on a public dataset of 20,275 high-resolution images across 10 defect classes, the model achieved strong performance with 92.7% accuracy, 97% recall, and real-time processing at 40 fps. These results demonstrate improved generalization and precision, especially for subtle and small-scale defects, compared to state-of-the-art models such as YOLOv8 and EfficientDet.
The integration of BRO enhanced feature selection efficiency, while the multi-attention mechanism captured complex spatial relationships, making the model suitable for smart manufacturing applications aligned with Industry 4.0 objectives. Despite its strengths, limitations include high computational cost, sensitivity to noise and lighting, and reduced adaptability to unseen wood types. To address these, we recommend future work on model compression (such as pruning, quantization, and distillation), domain adaptation, real-time video integration, and support for 3D imaging. A phased implementation roadmap is proposed: (1) model optimization, (2) dataset expansion with additional wood species, (3) deployment on edge hardware (such as Jetson Nano), and (4) real-world validation in industrial settings. These steps aim to ensure scalability, robustness, and practical value of the proposed system for intelligent visual inspection pipelines.
Footnotes
Acknowledgements
The authors are extremely grateful to Ton Duc Thang University and Van Lang University, Vietnam, for their support of this research.
Ethical considerations
This study was conducted in accordance with all relevant ethical guidelines and regulations.
Consent to participate
All participants involved in this study provided their voluntary and informed consent before participation.
Consent for publication
The authors confirm that all participants provided explicit consent for their data and information to be published in this article.
Author contributions
Conceptualization, Minh Ly Duc (M.L.D.); Nguyen Thi Phuong Thao (N.T.P.T.); methodology, M.L.D.; N.T.P.T.; software, M.L.D.; validation, M.L.D.; N.T.P.T.; formal analysis, M.L.D.; N.T.P.T.; investigation, M.L.D.; N.T.P.T.; resources, M.L.D.; data curation, M.L.D.; N.T.P.T.; Nguyen Quang Sang (N.Q.S.); writing—original draft preparation, M.L.D.; N.T.P.T.; writing—review and editing, M.L.D.; N.T.P.T.; N.Q.S.; visualization, M.L.D.; Supervision, M.L.D.; N.T.P.T.; N.Q.S.; project administration, M.L.D.; N.T.P.T.; All authors have read and agreed to the published version of the manuscript.
Funding
The authors disclosed receipt of the following financial support for the research, authorship, and/or publication of this article: This research was funded by the Van Lang University, Vietnam, under grant number VLU-2604-DT-KTM-GV-0075.
Declaration of conflicting interests
The authors declared the following potential conflicts of interest with respect to the research, authorship, and/or publication of this article: Data Availability Statement: All data supporting the findings of this study are available from the corresponding author upon reasonable request.
Data availability statement
All data supporting the findings of this study are available from the corresponding author upon reasonable request.
