Figure 1.
Overall experimental methodology pipeline, spanning dataset preprocessing, baseline YOLOv26n training, attention module integration, Optuna TPE hyperparameter optimization, optimized retraining, and multi-metric evaluation.
Figure 1.
Overall experimental methodology pipeline, spanning dataset preprocessing, baseline YOLOv26n training, attention module integration, Optuna TPE hyperparameter optimization, optimized retraining, and multi-metric evaluation.
Figure 2.
Representative sample image of each of the 22 dental surgical instrument classes, labeled with the full class name and the abbreviation code used in the class-level analysis (e.g., the confusion matrix,
Figure 3). Each image shows a single instrument on a surgical drape under the varied illumination and viewpoints that characterize the dataset.
Figure 2.
Representative sample image of each of the 22 dental surgical instrument classes, labeled with the full class name and the abbreviation code used in the class-level analysis (e.g., the confusion matrix,
Figure 3). Each image shows a single instrument on a surgical drape under the varied illumination and viewpoints that characterize the dataset.
Figure 4.
YOLO26n architecture, redrawn based on the official YOLO26 release [
35]. Purple and amber arrows mark the backbone (C3k2) and neck-output (layers 16/19/22) attention-insertion points of the eight variants; inference is end-to-end and NMS-free.
Figure 4.
YOLO26n architecture, redrawn based on the official YOLO26 release [
35]. Purple and amber arrows mark the backbone (C3k2) and neck-output (layers 16/19/22) attention-insertion points of the eight variants; inference is end-to-end and NMS-free.
Figure 5.
Block diagrams of the evaluated attention mechanisms: (a) Squeeze-and-Excitation (SE), (b) Convolutional Block Attention Module (CBAM), (c) Efficient Channel Attention (ECA), and (d) Simple Attention Module (SimAM). Each diagram shows the input tensor, the intermediate operations, and the resulting attention-modulated output.
Figure 5.
Block diagrams of the evaluated attention mechanisms: (a) Squeeze-and-Excitation (SE), (b) Convolutional Block Attention Module (CBAM), (c) Efficient Channel Attention (ECA), and (d) Simple Attention Module (SimAM). Each diagram shows the input tensor, the intermediate operations, and the resulting attention-modulated output.
Figure 6.
Comparison of backbone-level versus neck-level attention integration strategies. Neck-level attention preserves pretrained backbone weights and yields competitive performance (ECA-Neck-Opt: mAP@50 = 0.9592) with a lower transfer-learning risk.
Figure 6.
Comparison of backbone-level versus neck-level attention integration strategies. Neck-level attention preserves pretrained backbone weights and yields competitive performance (ECA-Neck-Opt: mAP@50 = 0.9592) with a lower transfer-learning risk.
Figure 7.
Ablation study of standard versus Optuna-optimized model variants: (a) mAP@50, (b) mAP@50–95, and (c) F1-score. Each bar pair compares the standard (blue) and Optuna-Opt (green) configuration of the same architecture.
Figure 7.
Ablation study of standard versus Optuna-optimized model variants: (a) mAP@50, (b) mAP@50–95, and (c) F1-score. Each bar pair compares the standard (blue) and Optuna-Opt (green) configuration of the same architecture.
Figure 8.
Optuna TPE optimization results: (a) best mAP@50 trajectory across 30 trials; (b) learning-rate versus mAP@50 scatter colored by batch size.
Figure 8.
Optuna TPE optimization results: (a) best mAP@50 trajectory across 30 trials; (b) learning-rate versus mAP@50 scatter colored by batch size.
Figure 9.
Absolute Pearson correlation of each hyperparameter with mAP@50 across the 30 TPE trials. HSV-V augmentation strength and the initial learning rate (lr0) exhibit the strongest influence on detection performance.
Figure 9.
Absolute Pearson correlation of each hyperparameter with mAP@50 across the 30 TPE trials. HSV-V augmentation strength and the initial learning rate (lr0) exhibit the strongest influence on detection performance.
Figure 10.
Validation mAP@50 (left) and validation box loss (right) per epoch for the standard model variants.
Figure 10.
Validation mAP@50 (left) and validation box loss (right) per epoch for the standard model variants.
Figure 11.
Validation mAP@50 (left) and training box loss (right) per epoch for the Optuna-optimized model variants.
Figure 11.
Validation mAP@50 (left) and training box loss (right) per epoch for the Optuna-optimized model variants.
Figure 12.
Per-architecture Optuna optimization gain (mAP@50 and mAP@50–95). SimAM exhibits the largest absolute improvement; SE shows the only negative change.
Figure 12.
Per-architecture Optuna optimization gain (mAP@50 and mAP@50–95). SimAM exhibits the largest absolute improvement; SE shows the only negative change.
Figure 13.
Precision (a) and recall (b) for every model variant. Blue: standard training; green: Optuna-optimized.
Figure 13.
Precision (a) and recall (b) for every model variant. Blue: standard training; green: Optuna-optimized.
Figure 14.
Radar comparison of the Optuna-optimized variants across five metrics (mAP@50, mAP@50–95, precision, recall, and F1). CBAM-Opt and ECA-Neck-Opt occupy the largest radar areas.
Figure 14.
Radar comparison of the Optuna-optimized variants across five metrics (mAP@50, mAP@50–95, precision, recall, and F1). CBAM-Opt and ECA-Neck-Opt occupy the largest radar areas.
Figure 15.
Metric-improvement heatmap (percentage-point change of each Optuna-optimized variant relative to its standard counterpart). Green: improvement; red: degradation. SimAM exhibits the most consistent gains across metrics.
Figure 15.
Metric-improvement heatmap (percentage-point change of each Optuna-optimized variant relative to its standard counterpart). Green: improvement; red: degradation. SimAM exhibits the most consistent gains across metrics.
Figure 16.
Per-class precision–recall curves for the CBAM-Opt model. The blue curve summarizes model performance over all classes (mAP@0.5 = 0.9593).
Figure 16.
Per-class precision–recall curves for the CBAM-Opt model. The blue curve summarizes model performance over all classes (mAP@0.5 = 0.9593).
Figure 17.
Sample CBAM-Opt prediction results on the dental instrument dataset, showing the detected instrument class, bounding box, and confidence score for each image.
Figure 17.
Sample CBAM-Opt prediction results on the dental instrument dataset, showing the detected instrument class, bounding box, and confidence score for each image.
Figure 18.
Accuracy–efficiency trade-off bubble plot (bubble area proportional to parameter count; ∘: standard; ⋄: Optuna-optimized). Attention modules deliver meaningful mAP gains with negligible (<5%) FLOPs overhead.
Figure 18.
Accuracy–efficiency trade-off bubble plot (bubble area proportional to parameter count; ∘: standard; ⋄: Optuna-optimized). Attention modules deliver meaningful mAP gains with negligible (<5%) FLOPs overhead.
Figure 19.
Seed-level distribution of mAP@50 and mAP@50–95 across five seeds per optimized configuration (validation and test splits). Individual seed values are overlaid on each box.
Figure 19.
Seed-level distribution of mAP@50 and mAP@50–95 across five seeds per optimized configuration (validation and test splits). Individual seed values are overlaid on each box.
Table 1.
Optuna TPE hyperparameter search space.
Table 1.
Optuna TPE hyperparameter search space.
| Parameter | Range | Description | Type |
|---|
| lr0 | Log-Uniform
[1 × , 1 × ] | Initial learning rate | Continuous |
| momentum | Uniform [0.70, 0.99] | AdamW | Continuous |
| weight_decay | Log-Uniform [1 × , 1 × ] | L2 regularization | Continuous |
| batch | {16, 32, 48, 64} | Batch size | Discrete |
| warmup_epochs | Uniform [1.0, 5.0] | Warmup duration | Continuous |
| hsv_s | Uniform [0.0, 1.0] | Saturation augmentation | Continuous |
| hsv_v | Uniform [0.0, 1.0] | Value augmentation | Continuous |
| translate | Uniform [0.0, 0.2] | Translation augmentation | Continuous |
| scale | Uniform [0.0, 0.9] | Scale augmentation | Continuous |
| fliplr | Uniform [0.0, 1.0] | Horizontal flip prob. | Continuous |
| mixup | Uniform [0.0, 0.3] | MixUp probability | Continuous |
| copy_paste | Uniform [0.0, 0.3] | Copy-Paste probability | Continuous |
| label_smoothing | Uniform [0.0, 0.1] | Label smoothing eps | Continuous |
Table 2.
Optuna TPE search space and the single shared configuration (trial #24) applied to all optimized variants, shown alongside the remaining top-five trials of the study. A single study was run on the baseline; no per-architecture re-optimization was performed. Learning rate and weight decay are given in scientific notation.
Table 2.
Optuna TPE search space and the single shared configuration (trial #24) applied to all optimized variants, shown alongside the remaining top-five trials of the study. A single study was run on the baseline; no per-architecture re-optimization was performed. Learning rate and weight decay are given in scientific notation.
| Parameter | Search Range | Selected (#24) | #23 | #27 | #11 | #21 |
|---|
| lr0 | Log-U[1 × , 1 × ] | 2.44 × | 2.58 × | 3.13 × | 1.29 × | 1.82 × |
| momentum | U[0.70, 0.99] | 0.9267 | 0.917 | 0.9461 | 0.8913 | 0.9136 |
| weight_decay | Log-U[1 × , 1 × ] | 1.96 × | 5.51 × | 1.31 × | 1.15 × | 1.00 × |
| batch | {16, 32, 48, 64} | 48 | 48 | 48 | 48 | 48 |
| warmup_epochs | U[1.0, 5.0] | 3.6599 | 3.6641 | 3.626 | 3.7568 | 3.8715 |
| hsv_s | U[0.0, 1.0] | 0.4112 | 0.4883 | 0.542 | 0.7078 | 0.6864 |
| hsv_v | U[0.0, 1.0] | 0.6573 | 0.7426 | 0.6836 | 0.9489 | 0.991 |
| translate | U[0.0, 0.2] | 0.1381 | 0.1427 | 0.1885 | 0.1165 | 0.1269 |
| scale | U[0.0, 0.9] | 0.0844 | 0.0051 | 0.2646 | 0.0067 | 0.0181 |
| fliplr | U[0.0, 1.0] | 0.0688 | 0.2044 | 0.082 | 0.0188 | 0.0134 |
| mixup | U[0.0, 0.3] | 0.1306 | 0.1238 | 0.2182 | 0.1574 | 0.1676 |
| copy_paste | U[0.0, 0.3] | 0.1223 | 0.1231 | 0.2149 | 0.07 | 0.0873 |
| label_smoothing | U[0.0, 0.1] | 0.0429 | 0.0449 | 0.0533 | 0.0446 | 0.0449 |
Table 3.
Standard training hyperparameter set used by all model variants.
Table 3.
Standard training hyperparameter set used by all model variants.
| Hyperparameter | Value |
|---|
| Image size | 640 × 640 px |
| Batch size | 32 |
| Epochs | 100 (patience = 10) |
| Optimizer | AdamW |
| Initial learning rate (lr0) | 0.01 |
| Momentum () | 0.937 |
| Weight decay | 5 × |
| Warmup epochs | 3 |
| Box loss gain | 7.5 |
| Classification loss gain | 0.5 |
| DFL loss gain | 1.5 |
| Mosaic augmentation | 1.0 |
| HSV-Saturation | 0.7 |
| HSV-Value | 0.4 |
| Horizontal flip | 0.5 |
| Mixed precision | FP16 |
Table 4.
Validation metrics under standard hyperparameters (green-shaded cells mark the highest value per metric).
Table 4.
Validation metrics under standard hyperparameters (green-shaded cells mark the highest value per metric).
| Model | mAP@50 | mAP@50–95 | Precision | Recall | F1 |
|---|
| Base | 0.9400 | 0.7762 | 0.8766 | 0.9008 | 0.8851 |
| SE | 0.9367 | 0.7756 | 0.9185 | 0.8846 | 0.8999 |
| CBAM | 0.9371 | 0.7663 | 0.9204 | 0.8718 | 0.8893 |
| ECA | 0.9383 | 0.7738 | 0.8817 | 0.8944 | 0.8860 |
| SimAM | 0.9155 | 0.7370 | 0.8620 | 0.8549 | 0.8525 |
| SPPF+ | 0.9305 | 0.7679 | 0.9004 | 0.8659 | 0.8789 |
| ECA-Neck | 0.9388 | 0.7741 | 0.9021 | 0.8919 | 0.8952 |
| CBAM-Neck | 0.9260 | 0.7518 | 0.8784 | 0.8784 | 0.8815 |
Table 5.
Validation metrics of Optuna-optimized variants (mAP@50 relative to standard). Green-shaded cells mark the highest value in each column; the red-shaded cell marks the only negative mAP@50.
Table 5.
Validation metrics of Optuna-optimized variants (mAP@50 relative to standard). Green-shaded cells mark the highest value in each column; the red-shaded cell marks the only negative mAP@50.
| Model | mAP@50 | mAP@50–95 | Precision | Recall | F1 | mAP@50 |
|---|
| Base-Opt | 0.9576 | 0.8000 | 0.9221 | 0.8933 | 0.9055 | +0.0176 |
| SE-Opt | 0.9237 | 0.7291 | 0.8802 | 0.8441 | 0.8589 | −0.0130 |
| CBAM-Opt | 0.9593 | 0.7991 | 0.9288 | 0.9002 | 0.9128 | +0.0222 |
| ECA-Opt | 0.9485 | 0.7938 | 0.8886 | 0.9095 | 0.8973 | +0.0102 |
| SimAM-Opt | 0.9553 | 0.7870 | 0.9187 | 0.8987 | 0.9061 | +0.0398 |
| SPPF+-Opt | 0.9500 | 0.7933 | 0.9146 | 0.8830 | 0.8970 | +0.0195 |
| ECA-Neck-Opt | 0.9592 | 0.7961 | 0.9177 | 0.8941 | 0.9040 | +0.0204 |
| CBAM-Neck-Opt | 0.9525 | 0.7902 | 0.9094 | 0.8984 | 0.9025 | +0.0265 |
Table 6.
Repeated-seed statistical validation of the optimized models (seeds 0, 1, 2, 3, 42): mean ± standard deviation and 95% confidence interval over five independent runs. Italic subheadings separate the validation and test blocks.
Table 6.
Repeated-seed statistical validation of the optimized models (seeds 0, 1, 2, 3, 42): mean ± standard deviation and 95% confidence interval over five independent runs. Italic subheadings separate the validation and test blocks.
| Model | mAP@50 (Mean ± SD) | mAP@50 95% CI | mAP@50–95 | Precision | Recall | F1 |
|---|
| val split
|
| Base-Opt | 0.9527 ± 0.0084 | [0.9423, 0.9631] | 0.7900 ± 0.0158 | 0.9180 | 0.8912 | 0.9020 |
| SE-Opt | 0.9455 ± 0.0127 | [0.9297, 0.9613] | 0.7778 ± 0.0293 | 0.9043 | 0.8840 | 0.8915 |
| CBAM-Opt | 0.9522 ± 0.0109 | [0.9387, 0.9657] | 0.7899 ± 0.0193 | 0.9080 | 0.9007 | 0.9021 |
| ECA-Neck-Opt | 0.9555 ± 0.0051 | [0.9491, 0.9618] | 0.7888 ± 0.0141 | 0.9113 | 0.9119 | 0.9098 |
| SimAM-Opt | 0.9429 ± 0.0144 | [0.9250, 0.9608] | 0.7689 ± 0.0224 | 0.9114 | 0.8895 | 0.8973 |
| test split |
| Base-Opt | 0.9208 ± 0.0116 | [0.9064, 0.9352] | 0.7552 ± 0.0175 | 0.8757 | 0.8970 | 0.8850 |
| SE-Opt | 0.9195 ± 0.0104 | [0.9065, 0.9325] | 0.7460 ± 0.0243 | 0.8717 | 0.8955 | 0.8820 |
| CBAM-Opt | 0.9264 ± 0.0120 | [0.9115, 0.9413] | 0.7598 ± 0.0228 | 0.8827 | 0.9019 | 0.8904 |
| ECA-Neck-Opt | 0.9283 ± 0.0170 | [0.9072, 0.9494] | 0.7585 ± 0.0240 | 0.8814 | 0.9018 | 0.8902 |
| SimAM-Opt | 0.9169 ± 0.0169 | [0.8959, 0.9378] | 0.7466 ± 0.0295 | 0.8661 | 0.9014 | 0.8823 |
Table 7.
Paired significance tests between the top optimized configurations. The paired t-test is the primary test; because the exact two-sided Wilcoxon signed-rank test cannot fall below at , it is reported as a direction-consistency check.
Table 7.
Paired significance tests between the top optimized configurations. The paired t-test is the primary test; because the exact two-sided Wilcoxon signed-rank test cannot fall below at , it is reported as a direction-consistency check.
| Split | Metric | Comparison | Mean Diff. | t | p (t) | Sig. | p (W) |
|---|
| val | mAP@50 | CBAM-Opt vs. ECA-Neck-Opt | −0.0033 | −0.666 | 0.5417 | no | 1.0 |
| val | mAP@50–95 | CBAM-Opt vs. ECA-Neck-Opt | +0.0010 | 0.101 | 0.9242 | no | 1.0 |
| val | mAP@50 | CBAM-Opt vs. Base-Opt | −0.0005 | −0.086 | 0.9355 | no | 1.0 |
| val | mAP@50–95 | CBAM-Opt vs. Base-Opt | −0.0002 | −0.015 | 0.9888 | no | 1.0 |
| val | mAP@50 | CBAM-Opt vs. SE-Opt | +0.0067 | 0.785 | 0.4765 | no | 0.625 |
| val | mAP@50–95 | CBAM-Opt vs. SE-Opt | +0.0120 | 0.668 | 0.5408 | no | 1.0 |
| test | mAP@50 | CBAM-Opt vs. ECA-Neck-Opt | −0.0019 | −0.238 | 0.8232 | no | 0.625 |
| test | mAP@50–95 | CBAM-Opt vs. ECA-Neck-Opt | +0.0012 | 0.109 | 0.9182 | no | 0.8125 |
| test | mAP@50 | CBAM-Opt vs. Base-Opt | +0.0056 | 0.645 | 0.5538 | no | 0.4375 |
| test | mAP@50–95 | CBAM-Opt vs. Base-Opt | +0.0046 | 0.291 | 0.7858 | no | 0.625 |
| test | mAP@50 | CBAM-Opt vs. SE-Opt | +0.0069 | 0.955 | 0.3939 | no | 0.4375 |
| test | mAP@50–95 | CBAM-Opt vs. SE-Opt | +0.0137 | 0.809 | 0.4638 | no | 0.4375 |
Table 8.
Computational cost of each attention configuration. C denotes the channel count at the integration point, and r is the reduction ratio; the percentage columns express the extra cost relative to the Base YOLOv26n model (2.60M parameters/8.2 GFLOPs).
Table 8.
Computational cost of each attention configuration. C denotes the channel count at the integration point, and r is the reduction ratio; the percentage columns express the extra cost relative to the Base YOLOv26n model (2.60M parameters/8.2 GFLOPs).
| Configuration | Extra Parameters (Theoretical, Per Block) | Extra Parameters (% of Base) | Extra FLOPs (% of Base) | Note |
|---|
| Base (YOLOv26n) | —(reference: 2.60M total params) | —(8.2 GFLOPs total) | — | Reference baseline (no attention) |
| + SE () | | ∼+1.0% | ∼+1.0% | Channel-only re-weighting |
| + CBAM () | | ∼+1.5% | ∼+1.5% | Channel + 7 × 7 spatial mask |
| + ECA | , | <+0.5% | ∼+0.5% | 1D conv; no dim. reduction |
| + SimAM | 0 (parameter-free) | 0 | ∼0% | Analytical 3D attention |
| + SPPF+ | ≈150 K (three pool branches) | ∼+5.8% | ∼+5.0% | Multi-kernel SPPF (k = 3,5,7) |
| + ECA-Neck (3 scales) | | <+0.5% | ∼+1.0% | Applied at P3/P4/P5 of neck |
| + CBAM-Neck (3 scales) | | ∼+1.5% | ∼+1.5% | Applied at P3/P4/P5 of neck |
Table 9.
Inference speed comparison (NVIDIA GPU, FP16; approximate values).
Table 9.
Inference speed comparison (NVIDIA GPU, FP16; approximate values).
| Model | Preprocess (ms) | Inference (ms) | Postprocess (ms) |
|---|
| Base | 3.1 | 7.6 | 2.7 |
| SE | 3.1 | 7.7 | 2.7 |
| CBAM | 3.2 | 7.8 | 2.8 |
| ECA | 3.1 | 7.7 | 2.7 |
| SimAM | 3.1 | 7.6 | 2.7 |
| SPPF+ | 3.2 | 8.1 | 2.8 |
| ECA-Neck | 3.1 | 7.8 | 2.7 |
| CBAM-Neck | 3.2 | 7.9 | 2.8 |
Table 10.
Estimated training time and attention module overhead.
Table 10.
Estimated training time and attention module overhead.
| Model | Epoch (s, est.) | Total (Min) | Overhead | Note |
|---|
| Base | ∼3.5 | ∼210 | — | Reference |
| SE | ∼3.6 | ∼216 | +∼3% | Channel attention |
| CBAM | ∼3.7 | ∼222 | +∼6% | Channel + spatial |
| ECA | ∼3.6 | ∼216 | +∼3% | 1D conv |
| SimAM | ∼3.5 | ∼210 | ∼0 | Parameter-free |
| SPPF+ | ∼3.8 | ∼228 | +∼9% | 3-stream pooling |
| ECA-Neck | ∼3.7 | ∼222 | +∼6% | 3-scale ECA |
| CBAM-Neck | ∼3.8 | ∼228 | +∼9% | 3-scale CBAM |
| Base-Opt | ∼3.5 | ∼210 | +Opt: 8–10 s × 30 | Single Optuna study (baseline) |
| CBAM-Opt | ∼3.7 | ∼222 | — | Retrained with shared best config |
Table 11.
Quantitative comparison with selected literature (bold, green-shaded rows denote this work).
Table 11.
Quantitative comparison with selected literature (bold, green-shaded rows denote this work).
| Study | Year | Model | Attention | HPO | Data | mAP@50/ mAP@50–95 | Note |
|---|
| Redmon et al. [1] | 2016 | YOLO | — | — | General | — | Slow training |
| Liu et al. [25] | 2024 | YOLOv7-Rep | — | — | Laparoscopic | — | Embedded |
| Jiang et al. [23] | 2023 | YOLOv5+ | Custom | — | Laparoscopic | ∼0.91 AP/— | Few classes |
| Ran et al. [24] | 2023 | YOLOv7x | Custom | — | Laparoscopic | ∼0.89 mAP/— | Embedded focus |
| Chien et al. [13] | 2024 | YOLOv8-AM | CBAM/ECA | — | X-ray | 0.94+ /— | Single attention |
| Wang et al. [22] | 2021 | YOLOv4 | — | — | Surgical | ∼0.88/— | Older architecture |
| Poomrittigul [6] | 2026 | YOLOv8 | — | — | Dental | —/— | First dental YOLO |
| This work (CBAM-Opt)
| 2026 | YOLOv26n | SE + CBAM + ECA + SimAM + Neck | Optuna-TPE | Dental (22 cls.) | 0.9593/0.7991 | Highest mAP@50 |
| This work (Base-Opt) | 2026 | YOLOv26n | — | Optuna-TPE | Dental (22 cls.) | 0.9576/0.8000 | Highest mAP@50–95 |
Table 12.
Original random split vs. group-disjoint, class-stratified 70/20/10 re-split. Base-Opt and CBAM-Opt were retrained from scratch on the re-split. Precision/Recall cells show original/group-disjoint values.
Table 12.
Original random split vs. group-disjoint, class-stratified 70/20/10 re-split. Base-Opt and CBAM-Opt were retrained from scratch on the re-split. Precision/Recall cells show original/group-disjoint values.
| Model | Split | mAP@50 Orig. | mAP@50 Group | mAP@50–95 Orig. | mAP@50–95 Group | P (Orig/Grp) | R (Orig/Grp) |
|---|
| Base-Opt | val | 0.9576 | 0.8865 | 0.8000 | 0.6892 | 0.9221/0.8526 | 0.8933/0.8421 |
| Base-Opt | test | 0.9224 | 0.9312 | 0.7591 | 0.7271 | 0.8679/0.8426 | 0.9093/0.8952 |
| CBAM-Opt | val | 0.9593 | 0.8936 | 0.7991 | 0.7100 | 0.9288/0.8611 | 0.9002/0.8537 |
| CBAM-Opt | test | 0.9350 | 0.9593 | 0.7731 | 0.7643 | 0.8921/0.9200 | 0.9191/0.9064 |