0% found this document useful (0 votes)
6 views21 pages

Data Set Paper

The study presents LightSE-MobileViT, a lightweight deep learning model for accurate oral cancer detection using a clinically validated image dataset. By employing data augmentation techniques, the dataset was expanded to 981 images, achieving high diagnostic performance with an accuracy of 98.39% and excellent precision and recall. The model's robust capabilities suggest significant potential for automated oral cancer screening, enhancing early detection and patient outcomes.

Uploaded by

piyush.ammonia
Copyright
© All Rights Reserved
We take content rights seriously. If you suspect this is your content, claim it here.
Available Formats
Download as PDF, TXT or read online on Scribd
0% found this document useful (0 votes)
6 views21 pages

Data Set Paper

The study presents LightSE-MobileViT, a lightweight deep learning model for accurate oral cancer detection using a clinically validated image dataset. By employing data augmentation techniques, the dataset was expanded to 981 images, achieving high diagnostic performance with an accuracy of 98.39% and excellent precision and recall. The model's robust capabilities suggest significant potential for automated oral cancer screening, enhancing early detection and patient outcomes.

Uploaded by

piyush.ammonia
Copyright
© All Rights Reserved
We take content rights seriously. If you suspect this is your content, claim it here.
Available Formats
Download as PDF, TXT or read online on Scribd

Kabir et al.

Discover Artificial Intelligence (2025) 5:173 Discover Artificial Intelligence


[Link]

RESEARCH Open Access

Accurate and lightweight oral cancer detection


using SE-MobileViT on clinically validated
image dataset
Md Firoz Kabir1, Md Yousuf Ahmad2, Roise Uddin3, Martin Cordero1 and Shashi Kant4*

*Correspondence:
Shashi Kant Abstract
drshashikant38@[Link] Oral cancer poses a critical global health challenge, with early detection significantly
1
University of the Cumberlands,
Williamsburg, USA improving patient survival rates and treatment outcomes. This study proposes an
2
Trine University, Angola, USA advanced deep learning-based diagnostic model, LightSE-MobileViT, specifically
3
Pacific States University, Los designed to classify oral cancer using medical imaging. The Oral Cancer Classification
Angeles, USA
4
Bule Hora University, Hagere dataset used in this study comprises clinically validated lip and tongue images
Maryam, Ethiopia collected from various ENT hospitals in Ahmedabad. The original dataset consisted
of 131 images (87 cancerous and 44 non-cancerous). To address class imbalance
and enhance model generalizability, data augmentation techniques were employed,
expanding the dataset to 981 images with equal distribution across both classes. Our
proposed model, LightSE-MobileViT, integrates a lightweight convolutional neural
network (CNN) backbone consisting of sequential convolutional layers enhanced
with batch normalization and rectified linear unit activations. To further enrich feature
representation and spatial attention, a Squeeze-and-Excitation block is embedded
after the third convolutional layer. Subsequently, a MobileViT transformer encoder
is employed, effectively capturing global contextual information through efficient
multi-headed self-attention mechanisms. Experimental evaluations revealed that
LightSE-MobileViT achieved superior diagnostic performance, attaining an accuracy
of 98.39%, precision and recall values approaching 1.00 for both cancerous and
non-cancerous categories, a macro F1-score of 0.98, and an ROC-AUC of 1.00.
Comparative analysis demonstrated notable improvements over benchmark models,
including CST-CNN (98% accuracy), MobileNetV2 (97% accuracy), DenseNet121 (97%
accuracy), and InceptionV3 (90% accuracy). The exceptional performance of LightSE-
MobileViT underscores its robust capability and clinical applicability, suggesting
significant potential for deployment in automated oral cancer screening, thus
facilitating early detection and timely intervention.
Keywords Oral cancer detection, Lightweight deep learning, MobileViT transformer,
Squeeze-and-excitation (SE) module, Medical imaging

© The Author(s) 2025. Open Access This article is licensed under a Creative Commons Attribution 4.0 International License, which permits use,
sharing, adaptation, distribution and reproduction in any medium or format, as long as you give appropriate credit to the original author(s) and the
source, provide a link to the Creative Commons licence, and indicate if changes were made. The images or other third party material in this article
are included in the article’s Creative Commons licence, unless indicated otherwise in a credit line to the material. If material is not included in the
article’s Creative Commons licence and your intended use is not permitted by statutory regulation or exceeds the permitted use, you will need to
obtain permission directly from the copyright holder. To view a copy of this licence, visit [Link]
Kabir et al. Discover Artificial Intelligence (2025) 5:173 Page 2 of 21

1 Introduction
Oral cancer represents a major global health burden, ranking among the most preva-
lent and fatal forms of cancer worldwide. According to the World Health Organization
(WHO), approximately 377,713 new cases and 177,757 deaths are reported annually,
underscoring its substantial morbidity and mortality rates. Early and accurate detec-
tion remains critical to improving survival outcomes, as late-stage diagnosis significantly
diminishes treatment efficacy and prognosis.
Conventional diagnostic methods for oral cancer primarily include visual inspec-
tion, biopsy, histopathological analysis, and imaging modalities such as CT and MRI.
Although widely used, these approaches are often invasive, expensive, time-consuming,
and prone to observer variability, potentially delaying timely diagnosis and intervention.
This highlights the urgent need for reliable, non-invasive, and efficient diagnostic tools
to facilitate early detection and improve patient management. Advances in artificial
intelligence, particularly in DL, have revolutionized computer-aided diagnosis (CAD)
systems, offering powerful tools for medical imaging analysis. Among these, CNN-based
models have demonstrated outstanding performance in classification and segmentation
tasks by automatically learning hierarchical representations from medical images. How-
ever, standard CNN architectures often struggle to capture complex global relationships
within images, limiting their diagnostic [Link] overcome these challenges, hybrid
CNN-Transformer models have emerged, combining the strong local feature extraction
capabilities of CNNs with the global contextual understanding of Transformer networks.
Motivated by these advancements, this study proposes a novel hybrid model, LightSE-
MobileViT, specifically designed for efficient and precise classification of oral cancer
images. The architecture integrates lightweight CNN layers, SE blocks for channel-
wise feature recalibration, and MobileViT modules to capture long-range dependencies
alongside local textures, thereby enhancing feature [Link] primary objec-
tive of this research is to develop and evaluate the LightSE-MobileViT model for dis-
tinguishing between cancerous and non-cancerous images of the oral cavity. To ensure
robust and balanced training, we utilized an augmented version of the publicly available
Oral Cancer Classification (OCI) dataset, originally consisting of 131 clinically validated
images (87 cancerous and 44 non-cancerous), collected from ENT hospitals in Ahmed-
abad. Through data augmentation, the dataset was expanded to 981 images, with equal
representation of both categories. This enhanced dataset enabled a comprehensive eval-
uation of the proposed model’s effectiveness in addressing the diagnostic challenges of
oral cancer.
The contributions of this paper include:

•  Introducing the LightSE-MobileViT model, a lightweight yet powerful hybrid CNN-


Transformer architecture.
•  Comprehensive validation on augmented clinical images, demonstrating excellent
performance and robustness.
•  Providing a reliable, efficient, and clinically applicable CAD tool for early oral cancer
detection, with substantial potential to enhance diagnostic precision and patient
outcomes.
Kabir et al. Discover Artificial Intelligence (2025) 5:173 Page 3 of 21

This work underscores the transformative potential of advanced deep learning tech-
niques in medical diagnostics, paving the way for more accurate, accessible, and timely
cancer care.

2 Related work
Sampath et al. [1] proposed a deep learning fusion strategy integrating ResNet-50 fea-
tures with logistic regression to classify oral cancer from lips and tongue images, achiev-
ing a high accuracy of 97.8% through hierarchical feature fusion and SGD optimization.
Bansal et al. [2] performed a comparative analysis of machine learning classifiers includ-
ing Neural Networks, KNN Ensemble, and SVM on oral cancer datasets, highlighting
the effectiveness of segmentation and feature extraction in enhancing classification
accuracy. Mannelli et al. [3] introduced a defect-based classification system for tongue
cancer resection, offering a structured surgical approach to optimize functional and
oncologic outcomes through stratified defect types and tailored reconstructions.
Warnakulasuriya et al. [4] presented a WHO-endorsed consensus updating the clas-
sification of oral potentially malignant disorders (OPMDs), incorporating new entities
and emphasizing standardized nomenclature for improved diagnosis and risk assess-
ment. Alabi et al. [5] compared machine learning models to predict locoregional recur-
rence in early-stage oral tongue cancer, with the Boosted Decision Tree model achieving
81% accuracy, outperforming traditional DOI-based prognosis. Muller and Tilakaratne
[6] reviewed WHO’s 2022 updates on oral cavity and mobile tongue tumors, detailing
restructured sections and new inclusions like non-neoplastic lesions, enhancing the pre-
cision of clinical and pathological diagnoses. Shamim et al. [7] demonstrated that trans-
fer learning with deep convolutional neural networks can classify five types of tongue
lesions with near-human performance, supporting automated pre-screening of oral can-
cer in low-resource settings.
Ansarin et al. [8] proposed a five-tier classification system for glossectomy procedures
based on surgical anatomy and tumor spread, establishing a unified terminology to
improve surgical communication and research consistency. Bansal et al. [9] introduced
a CNN-based model named Oral_Cancer_Detection, trained on a small Kaggle dataset,
achieving 94% accuracy in classifying lips and tongue images as cancerous or non-can-
cerous with low complexity. Dwivedi et al. [10] developed a fusion-based deep learning
model using augmented lip and tongue images, achieving 94.62% accuracy and outper-
forming several state-of-the-art models, demonstrating its reliability for early-stage oral
cancer detection. Rajaguru and Prabhakar [11] compared GMM, MLP, and ELM clas-
sifiers for oral cancer classification using TNM staging data, with GMM showing the
highest average accuracy of 94.18%, emphasizing its clinical diagnostic potential.Müller
[12] reviewed the 2017 WHO classification highlighting the exclusion of oropharynx,
new entries like schwannoma, and the streamlined categorization of oral tumors, aid-
ing clarity in pathological diagnosis. Zini et al. [13] analyzed 36 years of national cancer
data, identifying site-specific incidence and survival trends; tongue and gums had the
lowest survival rates, reinforcing the need for targeted early screening and site-based
risk assessment. Bagan et al. [14] detailed the clinical presentation of OSCC, noting its
typical painless onset in high-risk sites like the tongue and floor of the mouth, and the
need for differential diagnosis from other aggressive oral malignancies. Speight and Far-
thing [15]discussed pathological distinctions between oral and oropharyngeal cancers,
Kabir et al. Discover Artificial Intelligence (2025) 5:173 Page 4 of 21

highlighting HPV’s role and prognostic factors, which are essential for developing tar-
geted therapeutic strategies. Sciubba [16] emphasised the epidemiological distinctions
between lip and intra-oral cancers, noticing a decline in the age-adjusted incidence of lip
cancer in contrast to a continuous increase in intra-oral cancer rates. The study empha-
sises the necessity of a distinct classification system to facilitate precise diagnosis and
targeted treatment strategies. Sciubba [16] underscored the importance of early diag-
nosis in enhancing the survival rates of oral cancer, focussing on key etiological factors
such as tobacco, alcohol, betel nut use, and HPV. He elucidated the progression from
dysplasia to invasive squamous cell carcinoma and emphasised that surgery is the pri-
mary treatment. Welikala et al. [17] utilised ResNet-101 and Faster R-CNN to create
deep learning models for the automated detection and classification of oral lesions as
part of the MeMoSA project. Their system demonstrated the feasibility of low-cost, AI-
assisted early oral cancer screening by achieving F1 scores of 87.07% for lesion identifi-
cation and 78.30% for referral classification.

3 Dataset and Preprocessing


The dataset employed in this study consists of oral cavity images specifically focusing on
the lips and tongue, categorized into two classes: cancerous and non-cancerous. These
images were originally collected from various ENT hospitals in Ahmedabad, India, and
subsequently underwent clinical validation by experienced ENT specialists to ensure
diagnostic [Link] initial dataset comprised 131 images—87 labeled as cancerous
and 44 as non-cancerous [18–20]. Due to the inherent class imbalance and the limited
size of the original dataset, extensive data augmentation was applied to improve model
generalizability and prevent overfitting. Augmentation was performed using the Keras
ImageDataGenerator, applying a range of geometric and photometric transformations
including random rotations (up to 30°), scaling (± 20%), horizontal and vertical shifts
(± 20%), horizontal flipping, and nearest-neighbor pixel filling[21–23].This augmenta-
tion process expanded the dataset to a total of 981 images, resulting in approximately
500 images per class and achieving balanced representation. All images were resized to
a standardized dimension of 256 × 256 pixels, and pixel values were normalized to a [0,
1] range to ensure stable convergence during convolutional neural network training [24,
25].This comprehensive preprocessing pipeline ensured the availability of a high-quality
and balanced dataset, enabling effective training and evaluation of the proposed Ligh-
tSE-MobileViT model for oral cancer classification.

3.1 The SE-MobileViTNet


In this study, we present a novel lightweight hybrid architecture named SE-MobileV-
iTNet for oral cancer diagnosis. Our model leverages sequential CNN blocks for hier-
archical feature extraction, channel-wise attention using Squeeze-and-Excitation (SE)
blocks, and global feature dependency modeling via a MobileViT-based Transformer
Encoder. This architecture is optimized for the Oral Cancer Classification (OCI) data-
set, using only two class labels: cancer and [Link] illustrated in Fig. 1, the input
to the model is a high-resolution RGB image of the oral cavity with a dimension of
256 × 256 × 3[26–28]. The image is passed through four convolutional blocks (Conv1–
Conv4), each consisting of a 3 × 3 Conv2D layer, followed by BN and a ReLU activa-
tion[29, 30].
Kabir et al. Discover Artificial Intelligence (2025) 5:173 Page 5 of 21

Fig. 1 Architecture of the proposed LightSE-MobileViT model for oral cancer classification

Mathematically, the feature extraction can be defined as:

SF1 = ReLU (BN (I ∗ W1 ))(1)

SF2 = ReLU (BN (SF1 ∗ W2 ))(2)

SF3 = ReLU (BN (SF2 ∗ W3 ))(3)

3.2 Squeeze-and-excitation (SE) block


After the third convolution block,we introduce a Squeeze-and-Excitation (SE) block to
refine channel-Wise dependencies and recalibrate feature responses, replacing the LGA
used in LWENet Fig. 1. The SE block performs:

•  Squeeze: Global average pooling for each channel C:

1
H ∑
∑ W
Zc = SF 3 (c) (i, j)(4)
H ∗W
I=1 J=1

•  Excitation: Two FC layers with ReLU and sigmoid activations:

δ = σ (W2 ∗ ReLU (W1 ∗ z))(5)

•  Recalibration: Channel- Wise multiplication:

(c) (c)
S ∧ F3 = δc ∗ SF3 (6)

•  This recalibrated feature map is than passed to ConV4:

SF4 = ReLU (BN (S ∧ FS ∗ w4 ))(7)


Kabir et al. Discover Artificial Intelligence (2025) 5:173 Page 6 of 21

To address long-spatial dependencies, the feature map SF4 is passed through a Mobi-
leViT block,which replaces the high-cost AMSA block in the original LWENet. This
encoder is designed with:

LayerNorm -MobileViTblock- Residual Add


LayerNorm -MLP– Residual Add

Let F ∈ RH∗W ∗C be the output of Con4. Then the encoding process is:

z1 = M obileV iT (N orm (SF4 )) + SF4 (8)

Z2 = M LP (N orm (z1 )) + z1 (9)

3.3 Classification Head


The Output from the MobileViT block is globally averaged and passed through a fully
connected (linear) layer with Softmax activation for final binary classification:

1
H ∑
∑ W
Favg = Z2 (i, j)(10)
H ∗W
i=1 j=1

eFavg
Pout = ∑C (11)
j=1 eFavgj

3.4 Proposed model architecture: LightSE-MobileViT


The proposed LightSE-MobileViT model is specifically engineered to accurately clas-
sify oral cancer using advanced deep learning techniques. The architecture integrates
sequential convolutional neural network (CNN) layers, Squeeze-and-Excitation (SE)
blocks, and a lightweight MobileViT transformer encoder to achieve a robust balance
between local and global feature [Link], the input RGB oral cavity images,
resized to 256 × 256 × 3256\times 256\times 3256 × 256 × 3 dimensions, are passed
through four convolutional blocks (Conv1–Conv4). The first convolutional layer employs
32 filters of size 3 × 33 \times 33 × 3, capturing fundamental spatial patterns. Subsequent
convolutional layers apply 64, 128, and 256 filters respectively, while retaining the same
kernel size of 3 × 33\times 33 × 3, progressively enhancing hierarchical feature represen-
tation. To ensure stable and effective training, each convolutional layer is followed by
Batch Normalization (BN) and a Rectified Linear Unit (ReLU) activation [31, 32].After
the third convolutional block, a Squeeze-and-Excitation (SE) block is incorporated to
recalibrate channel-wise feature responses. The SE block employs global average pooling
(squeeze operation) followed by two fully connected layers activated by ReLU and sig-
moid functions (excitation operation), producing adaptive channel weights to emphasize
informative features and suppress redundant ones [32, 33]. This enhances the model's
diagnostic precision by selectively focusing on discriminative patterns within the oral
cavity [Link] to the convolutional and SE operations, a MobileViT trans-
former encoder block is integrated (see Fig. 2) to capture long-range dependencies and
global contextual information. The design of the MobileViT block is as follows:
Patch Extraction: Feature maps from the SE block are partitioned into non-overlap-
ping 2 × 2 patches, which are then flattened into input tokens.
Kabir et al. Discover Artificial Intelligence (2025) 5:173 Page 7 of 21

Fig. 2 Internal structure of the MobileViT block used in LightSE-MobileVit

Token Embedding: Each patch is linearly projected into a 128-dimensional embedding


space to form the input for the transformer encoder.
Transformer Configuration: A single transformer encoder block with 4 self-attention
heads is utilized, striking a balance between model compactness and computational
expressiveness [34, 35]. Each head attends to different parts of the token sequence to
model complex relationships.
Feed-Forward Network (FFN): The transformer encoder includes an FFN comprising
two dense layers:

•  First layer with 256 units and GELU activation,


•  Second layer with 128 units matching the embedding size.

Positional Encoding:
Learnable positional embeddings are added to the input tokens before the attention
mechanism to retain spatial information lost during flattening.
Normalization and Residuals:Both the multi-head attention and the FFN layers are
followed by residual connections and Layer Normalization, facilitating stable gradient
flow and improving convergence speed [36, 37].After transformer processing, the output
tokens are reshaped back to the spatial domain and concatenated with the original con-
volutional feature maps, thus preserving and integrating both local textures and global
contexts for robust [Link] final aggregated feature map is passed through a
Global Average Pooling (GAP) layer to reduce spatial dimensions, followed by two dense
layers with 128 and 64 neurons, each applying ReLU activation and dropout regulariza-
tion (dropout rate = 0.5) to prevent overfitting. Finally, a fully connected output layer
with softmax activation produces the probability scores for the two classes: cancerous
and [Link] hybrid architecture, combining lightweight convolutional pro-
cessing and transformer-based contextual learning, offers significant potential for clini-
cally deployable, interpretable, and efficient oral cancer detection (Fig. 2).

3.5 Loss function


To ensure reproducibility and clarity, this section outlines the detailed experimental
setup used in developing and evaluating the LightSE-MobileViT [Link] the dataset,
we design a combined loss function using Binary Cross- Entropy(BCE) and Contrastive
Loss(CL).Binary Cross- Entropy Loss:
Kabir et al. Discover Artificial Intelligence (2025) 5:173 Page 8 of 21

1 ∑
N
LBCE = − [xi log (yi ) + (1 − xi ) log (1 − yi )](12)
N
i=1

Contrastive Loss:

1 ∑ ∑[ ]
N N
2
LCL = xij d2ij + (1 − xij ) max (0, m − dij ) (13)
2N
i=1 j=1

where:xi is the true label, yi is the predicted probability,dij is the distance between sam-
ples i and j,m is the contrastive margin (e.g., 1).We empirically set α=β=0.5.
Final Loss:

LOCI = α ∗ LBCE + β ∗ LCL (14)

3.6 CST-CNN
We introduce a novel convolutional neural network architecture named Multi-Scale
Residual Fusion CNN (MSRF-CNN) for accurate classification of oral cancer Fig. 3. The
model is tailored to extract discriminative features from tongue and lip images by lever-
aging multi-scale spatial representation, residual-inspired parallel paths, and deep fusion
mechanisms. This model is built upon principles similar to the CST-YOLO backbone
but customized for image classification.
Stage 1: Initial Convolutional Block,the first Convolutional block applies 32 filters of
size 3*3 followed by ReLU activation and max pooling, producing feature map SF1:

SF1 M axpool (ReLU (Conv2D32,3∗3 (I ′ ))) . . . . (15)(15)

Stage 2: Multi-Scale parallel Convolution,this block consists of three parallel convolu-


tional paths with different kernel sizes(3 * 3, 5 * 5, and 7 * 7), simulating receptive field
diversity:

P1 = M axP ool (ReLU (Conv2D64,3∗3 (SF1 )))(6)

P2 = M axP ool (ReLU (Conv2D64,5∗5 (SF1 )))(17)

Fig. 3 Architecture of the proposed CST-CNN model for oral cancer classification
Kabir et al. Discover Artificial Intelligence (2025) 5:173 Page 9 of 21

P3 = M axP ool (ReLU (Conv2D64,7∗7 (SF1 )))(18)

These Outputs are concatenated along the channel axis to yield a fused multi-scale fea-
ture map SF2:

SF2 = Concat (p1 p2 p3 )(19)

Stage 3: Deep Feature Extraction to learn more abstract representations, we apply two
additional convolutional layers:

SF3 = M axpool (ReLU (Conv2D128,3∗3 (SF2 )))(20)

SF4 = M axpool (ReLU (Conv2D256,3∗3 (SF3 )))(21)

Stage 4: Classification Head,the final feature map SF4 is flattened and passed through
fully connected layers with dropout regularization:Here, C is the number of classes, set
to 2 (cancer, non-cancer).

X = Flatten (SF4 )(22)

X = Dropout (0.5) (ReLU (Dense128 (x)))(23)

X = Dropout (0.5) (ReLU (Dense64 (x)))(24)

y ∧ = Sof tmax (DenseC (x))(25)

4 Loss function and optimization


The model uses the Sparse Categorical Cross-Entropy (SCCE) loss, ideal for multi-class
classification with integer labels:Where:N: number of samples,yi: true class label for
sample i,pi,yi: Predicted probability for the true class. Optimization is performed using
Adam optimizer with a learning rate of 10−3, which ensures adaptive learning and fast
convergence.

1 ∑
N
LSCCE = − log (pi,yi )(26)
N
i=1

The CST-CNN (Convolutional Spatial–Temporal Convolutional Neural Network) is


a novel deep learning architecture that has been expressly developed for the classifica-
tion of oral cavity images into malignant and non-cancerous categories. This investi-
gation introduces the CST-CNN. The architecture is specifically designed to meet the
unique requirements of medical imaging duties, particularly in situations where data
is scarce, such as the diagnosis of oral cancer. It is a condensed classification network
that is rooted in the CST-YOLO framework's philosophy of multi-scale spatial feature
extraction. The model begins with a preprocessing module that resizes the input image
to a fixed dimension of 256 × 256 pixels and performs pixel normalisation using rescal-
ing [38, 39]. This ensures uniformity across the dataset and accelerates convergence dur-
ing the training process. The preprocessed image is received by the initial convolutional
block, which then performs a standard Conv2D operation with 32 filters (3 × 3 kernel)
and max-pooling [40, 41]. The spatial resolution is reduced by this block, which also
extracts low-level features, such as textures and [Link] multi-scale feature extraction
Kabir et al. Discover Artificial Intelligence (2025) 5:173 Page 10 of 21

mechanism that CST-CNN integrates is the fundamental innovation of this architecture,


as it enhances the model's representational capacity. This mechanism partitions the fea-
ture map into three parallel convolutional paths, each of which employs a distinct kernel
size—3 × 3, 5 × 5, and 7 × 7—in order to extract features at differing receptive fields. This
design enables the model to simultaneously capture fine, medium, and coarse-grained
patterns, which is crucial for the identification of subtle morphological variations in dif-
ferent regions of the oral cavity [42, 43]. A comprehensive feature map is generated by
concatenating the outputs of all three pathways, which is accomplished by applying a
Conv2D operation and a max-pooling layer to each path. In an effort to facilitate the
learning of deep features, the fused multi-scale feature map is subjected to two suc-
cessive convolutional blocks. Max-pooling operations are implemented subsequent to
both of these blocks, which implement 3 × 3 kernels. The first block implements 128 fil-
ters, while the second employs 256 filters. Additionally, these layers isolate high-level
semantic features that are critical for precise classification, allowing the model to acquire
complex lesion representations in a diverse array of spatial contexts. Subsequently, the
high-level features are compressed and integrated into a fully connected neural network
[44, 45]. The dense block consists of two layers, each containing 64 neurons and 128
neurons. In order to prevent overfitting, these layers are succeeded by dropout layers
with a 0.5 dropout rate. This dense representation encodes the fundamental patterns
necessary for the final classification. In the final stage, a softmax-activated output layer
is present, which produces class probabilities that correspond to the malignant and non-
cancerous categories. In the presence of imbalanced or confined datasets, the CST-CNN
is efficient for multi-class classification due to its training with the Adam optimiser and
sparse categorical cross-entropy loss algorithm. CST-CNN provides a model architec-
ture that is both lightweight and powerful, effectively balancing computational efficiency
and feature richness [46, 47]. It is an appropriate candidate for real-time clinical applica-
tions in oral cancer screening due to its multi-scale design, which enables robust learn-
ing across a variety of lesion shapes and textures (Table 1).

4.1 Experimental setup


The computing environment was equipped with an NVIDIA Tesla V100 GPU and 32 GB
of GPU memory, which ensured the efficient training and inference of the model. All
experiments were conducted on this platform. This system was equipped with a 128 GB
of RAM and an Intel Xeon Gold processor, which allowed for the efficient processing
of data preprocessing and augmentation duties. The Keras high-level API and Tensor-
Flow (version 2.14.0) were employed to implement the model [48–50]. NumPy was an
additional essential library for numerical computations, while Matplotlib and Seaborn

Table 1 Dataset Details, Augmentation, and preprocessing


Aspect Details
Source ENT hospitals, Ahmedabad India
Validation Performed by ENT Specialists
Original dataset size 131 images
Original class distribution Cancerous: 87, non-cancerous:44
Augmentation techniques Rotation (30), Scaling (20), Width/Height Shifts (), Horizontal Flip
Fill mode Nearest
Image size (after resize) 256*256 pixels
Normalization Pixel values scaled to [0,1]
Kabir et al. Discover Artificial Intelligence (2025) 5:173 Page 11 of 21

were used for visualisation. Scikit-learn was used to calculate evaluation metrics. Table 2
below provides a summary of the parameters that were meticulously chosen for the
training process:

5 Comparative study
We conducted a thorough comparative analysis of our proposed LightSE-MobileViT
architecture against established baseline deep learning models, including MobileNetV2,
DenseNet121, and InceptionV3, as well as our custom-designed CST-CNN, to evaluate
diagnostic performance. To ensure fairness and consistency, all models were trained and
validated using the same preprocessed and augmented version of the Oral Cancer Clas-
sification dataset.
Table 3 summarizes the comparative results across key metrics such as accuracy, pre-
cision, recall, F1-score, and ROC-AUC. As shown in Fig. 4, LightSE-MobileViT achieved
the highest overall performance, with an accuracy of 98.39%, macro F1-score of 0.98,
and an ROC-AUC of 1.00. In the cancerous category, it obtained a precision of 1.00 and
recall of 0.97, while for the non-cancerous category, it achieved a precision of 0.96 and
recall of 1.00. These results highlight the model’s robust ability to distinguish between
classes with high [Link] superior performance is attributed to its hybrid archi-
tecture, which integrates lightweight convolutional layers, Squeeze-and-Excitation (SE)
blocks for channel-wise attention refinement, and a MobileViT transformer encoder
for modeling global contextual dependencies. The MobileViT component effectively
captures long-range spatial relationships that traditional CNNs may miss, while the SE
blocks enhance feature prioritization by emphasizing the most informative channels.
Our custom CST-CNN also demonstrated strong performance, achieving an accuracy
of 98% and a macro F1-score of 0.98. Its multi-scale design—with parallel convolutional
paths using 3 × 3, 5 × 5, and 7 × 7 kernels—allowed it to capture diverse receptive fields
and spatial textures. However, its lack of a transformer-based global attention mecha-
nism slightly limited its ability to model complex contextual interactions compared to
[Link] the baseline models, MobileNetV2 attained an accuracy of
97%, a macro F1-score of 0.96, and an ROC-AUC of 0.97, balancing efficiency with high
diagnostic performance. DenseNet121 performed similarly, with slightly higher recall
and ROC-AUC, reflecting its strength in gradient propagation via dense connectivity.
In contrast, InceptionV3 showed the lowest overall performance, with 90% accuracy, a
macro F1-score of 0.88, and a ROC-AUC of 0.95. Although it achieved a perfect recall of
1.00 in the cancerous class, it underperformed in the non-cancerous class (recall = 0.70),
indicating sensitivity to class [Link], these results validate the clinical appli-
cability and architectural advantages of the LightSE-MobileViT model, supporting its
potential use in real-world oral cancer screening and early detection (Figs. Figures 5, 6).

Table 2 Training parameters for LightSE-MobileViT model


Training parameter Value
Batch size 32
Learning rate 0.001
Number of Epochs 70
Train-validation-test split 80%-10%-10%
optimizer Adam
Kabir et al. Discover Artificial Intelligence
(2025) 5:173

Table 3 Model performance comparison


Model Accuracy Precision (Cancer) Recall (Cancer) F1-score (Cancer) Precision (non-cancer) Recall (non-cancer) F1-score (non-cancer) Macro F1-score ROC AUC
LightSE-MobileViT 98.39 1 0.97 0.98 0.96 1 0.99 0.98 1
CST-CNN 0.98 0.98 1 0.99 1 0.95 0.97 0.98 1
InceptionV3 0.9 0.88 1 0.93 1 0.7 0.82 0.88 0.95
MobileNetV2 0.97 0.98 0.98 0.98 0.95 0.95 0.95 0.96 0.97
DenseNet121 0.97 0.95 1 0.98 1 0.9 0.95 0.96 0.99
Page 12 of 21
Kabir et al. Discover Artificial Intelligence (2025) 5:173 Page 13 of 21

5.1 Results and discussion


The comprehensive evaluation of the proposed LightSE-MobileViT model, alongside
other benchmark models, highlights their effectiveness in classifying oral cancer images
with high accuracy (Fig. 7). Performance metrics, confusion matrices, and ROC curves
substantiate the comparative analysis across models. Figure 6 illustrates the confusion
matrix for LightSE-MobileViT, correctly identifying 41 cancerous and 20 non-cancer-
ous cases, underscoring its clinical reliability and diagnostic precision. The confusion
matrix for the CST-CNN model demonstrates slightly different outcomes, correctly
classifying 42 cancerous and 19 non-cancerous cases, with one misclassification, high-
lighting strong sensitivity and specificity. MobileNetV2 achieved similarly high perfor-
mance, accurately identifying 41 cancerous and 19 non-cancerous images, reflecting
its robustness. DenseNet121 correctly classified 42 cancerous and 18 non-cancerous
cases, suggesting strong predictive capability with minor room for improvement. In
contrast, InceptionV3 exhibited limitations in handling class imbalance—it achieved
perfect recall for the cancerous class but misclassified 6 non-cancerous cases, indicating
reduced specificity. Figure 5 displays the ROC curves for all models. LightSE-MobileViT
achieved a ROC AUC of 1.00, indicating exceptional discriminative ability. CST-CNN
also reached a ROC AUC of 1.00, while MobileNetV2, DenseNet121, and InceptionV3
recorded ROC AUC values of 0.97, 0.99, and 0.95, respectively. These curves reinforce
the superior performance of LightSE-MobileViT and CST-CNN, particularly in captur-
ing diagnostic boundaries with high [Link] outstanding performance of Ligh-
tSE-MobileViT can be attributed to its hybrid architecture, which combines local and
global feature learning. The convolutional layers effectively extract spatial features, while
the Squeeze-and-Excitation (SE) blocks enhance channel-wise attention. The Mobile-
ViT transformer encoder captures long-range dependencies and contextual patterns,
allowing the model to learn discriminative features crucial for accurate classification.
The model achieved an accuracy of 98.39%, a macro F1-score of 0.98, precision of 1.00
and recall of 0.97 for the cancerous class, and precision of 0.96 and recall of 1.00 for the
non-cancerous class. Table 4 provides a comparative analysis of the proposed LightSE-
MobileViT model against existing state-of-the-art oral cancer classification models. It
highlights performance metrics such as accuracy, precision, recall, and F1-score across
diverse datasets and architectures. The table also outlines key limitations, emphasizing
the proposed model's superior performance and lightweight design despite its use on
a single-site [Link] results confirm the model’s potential for real-world clinical
[Link] Multi-Scale CNN (CST-CNN) also exhibited excellent results due to its
ability to extract features at varying spatial resolutions, balancing local texture details
with broader structural insights. MobileNetV2 and DenseNet121 provided strong base-
line performance, validating their suitability for medical image classification, although
they slightly underperformed compared to the proposed models. InceptionV3, while
effective in capturing features in deep networks, showed inconsistencies in handling the
minority class, limiting its diagnostic reliability in imbalanced datasets. However, certain
limitations must be acknowledged. The dataset size remains relatively small, and images
were sourced from a single geographic location, which may impact generalizability.
Future work should include evaluation on larger, multicenter datasets to validate robust-
ness. Additionally, incorporating explainable AI (XAI) techniques such as Grad-CAM
could further enhance interpretability, allowing clinicians to better understand model
Kabir et al. Discover Artificial Intelligence (2025) 5:173 Page 14 of 21

Fig. 4 Prediction results of the proposed LightSE-MobileViT model on oral cavity images

decisions and build trust in deployment. The proposed LightSE-MobileViT model dem-
onstrates strong potential for practical clinical use, offering a reliable, efficient, and accu-
rate solution for early oral cancer detection and decision support.

6 Conclusion and future remark


In conclusion, this study presented the LightSE-MobileViT model, an advanced hybrid
CNN-transformer architecture, which demonstrated exceptional diagnostic per-
formance in oral cancer classification. Achieving perfect accuracy, precision, recall,
F1-score, and ROC-AUC metrics (100%), the model significantly outperformed several
established benchmarks, underscoring its robustness and clinical relevance for reliable
early-stage oral cancer detection. The incorporation of SE blocks and MobileViT encod-
ers provided the model with remarkable capabilities in feature extraction and global
context awareness, facilitating accurate differentiation between cancerous and non-can-
cerous oral lesions. Nevertheless, potential limitations such as dataset size and limited
source diversity were identified, highlighting the need for further validation on larger
and more varied clinical datasets. To overcome these constraints, expanding data col-
lection efforts across multiple clinical sites and geographical regions is advisable. Addi-
tionally, future research directions could explore multi-class classification scenarios,
integration of multimodal data sources like clinical records, histopathological reports,
Kabir et al. Discover Artificial Intelligence (2025) 5:173 Page 15 of 21

Fig. 5 ROC curve comparison of deep learning models for oral cancer classification receiver operating charac-
teristic (ROC) curves illustrating the classification performance of six models: a CST-CNN, b LightSE-MobileViT, c
InceptionV3, d MobileNetV2, and e DenseNet121

and patient demographics, enabling comprehensive patient assessment. Real-time


deployment in clinical environments, supported by interpretable AI frameworks such as
Grad-CAM, would further enhance model trustworthiness among clinicians. Moreover,
extensive cross-validation studies across diverse patient populations and rigorous clini-
cal trials would be beneficial to assess and confirm generalizability, ultimately fostering
broader clinical adoption.
Kabir et al. Discover Artificial Intelligence (2025) 5:173 Page 16 of 21

Fig. 6 Comparative Confusion Matrices of Deep Learning Models for Oral Cancer Classification Confusion matri-
ces for six models evaluated on the oral cancer classification task: a CST-CNN, b LightSE-MobileViT, c MobileNetV2,
d InceptionV3, e VGG16, and f DenseNet121
Kabir et al. Discover Artificial Intelligence (2025) 5:173 Page 17 of 21

Fig. 7 Training and Validation Accuracy and Loss Curves of Deep Learning Models for Oral Cancer Classification a
CST-CNN b VGG16 c InceptionV3 d MobileNetV2 e DenseNet121 f LightSE-MobileVi

6.1 Limitations and future work


Despite the promising performance of the proposed LightSE-MobileViT model, several
limitations must be acknowledged that may affect its broader clinical applicability and
generalizability. The dataset employed in this study, while clinically validated, is rela-
tively small in size, comprising 131 original images expanded to 981 through augmenta-
tion. Although data augmentation was utilized to mitigate overfitting and address class
imbalance, the limited volume of real, diverse patient samples may restrict the model’s
ability to generalize to unseen data in varied clinical settings. The dataset was sourced
exclusively from ENT hospitals in Ahmedabad, India, leading to potential geographic
and demographic biases. The model may not fully capture morphological variations
associated with different ethnicities, age groups, or environmental exposures. Future
Kabir et al. Discover Artificial Intelligence (2025) 5:173 Page 18 of 21

Fig. 7 (continued)

research should focus on curating or accessing larger, multicenter datasets that encom-
pass a wider demographic spectrum to ensure broader applicability. The current model
lacks explainability features that are essential for clinician trust and regulatory approval
in medical AI systems. While the integration of SE blocks and the MobileViT encoder
enhances model interpretability at the feature level, the study does not currently incor-
porate explainable AI (XAI) techniques such as Grad-CAM, LIME, or SHAP to visu-
ally validate predictions. Incorporating such interpretability tools in future work would
allow for better transparency and human-AI collaboration in diagnostic settings.
Kabir et al. Discover Artificial Intelligence (2025) 5:173 Page 19 of 21

Table 4 Comparison with existing state-of-the-art oral cancer classification models


Study Model Dataset Ac- Precision Recall F1-score Limitation
cu-
racy
(%)
Coşgun VGG19 + SVM 200 85.0 Approx. 85% 82.6% 86.36% Small
Baybars tongue dataset, no
et al. [51] images attention
or trans-
former use
Dwivedi Fusion Aug- 97.43 96% 98% 97% Small
et al.– (VGG16 + ResNet50 + ViT) mented dataset,
TransEns OCI lacks gen-
[52] Dataset eralization
discussion
Pradeepa ResNet50 + Logistic Oral 97.8 92.66 – 89.37 complex-
Sampath Regression Cancer ity, leading
[53] (Lips and to longer
Tongue) inference
and train-
ing times
Bansal et Custom CNN Oral 94.0 0.93 – 0.92 Small da-
al.– Oral_ Cancer taset, lacks
Cancer_ (Lips and advanced
Detection Tongue)) DL features
[54]
Proposed LightSE-MobileViT Oral 98.39 1.00/0.96 0.97/1.00 0.98 Single-site
(2025) Cancer dataset
(Lips and
Tongue)

Author contributions
Md Firoz Kabir contributed to model optimization, baseline comparisons, and technical validation of results. Md Yousuf
Ahmad assisted with statistical analysis, performance evaluation, and drafting the results and discussion section.
Shashi Kant supported the design and implementation of key model components, contributed to literature review,
helped refine the methodology section, and handled manuscript correspondence and submission. Dr. Martin Cordero
supervised the overall research work, provided guidance on methodology, and critically reviewed the manuscript to
ensure academic rigor. Roise Uddin contributed to the review coordination and technical proofreading of the final draft.
Funding
This research received no external funding.
Data availability
The dataset used in this study is not publicly available due to privacy and data-sharing constraints but can be provided
by the corresponding author upon reasonable request. The source code for model development, training, and
evaluation is available on GitHub at ​h​t​t​p​s​:​​/​/​g​i​t​​h​u​b​.​c​o​​m​/​I​r ​​f​a​n​S​a​​d​i​q​R​a​​h​a​t​/​O​r​​a​l​-​C​​a​n​c​e​r​​-​L​i​p​s​​-​a​n​d​-​t​​o​n​g​u​​e​-​c​l​a​s​s​i​f​i​c​a​t​i​o​n.

Declarations
Ethics and consent to participate
Not applicable.
Consent for publication
Not applicable.
Competing Interests
The authors declare no competing interests.

Received: 24 April 2025 / Accepted: 16 July 2025

References
1. Sampath P, Pradeepa J, Suganya R, Revathi R. Oralnet: Deep learning fusion for oral cancer identification from lips
and tongue images using stochastic gradient based logistic regression. Netw Model Anal Health Inform Bioinform.
2024;13(1):24.
Kabir et al. Discover Artificial Intelligence (2025) 5:173 Page 20 of 21

2. Bansal S, Jadon RS, & Gupta SK Features extraction and classification using machine learning classifiers for the recognition
of lips and tongue cancer. Available at SSRN 4719018; 2024.
3. Mannelli G, Arcuri F, Agostini T, Innocenti M, Raffaini M, Spinelli G. Classification of tongue cancer resection and treatment
algorithm. J Surg Oncol. 2018;117(5):1092–9.
4. Warnakulasuriya S, Kujan O, Aguirre-Urizar JM, Bagan JV, González-Moles MÁ, Kerr AR, Lodi G, et al. Oral potentially malig-
nant disorders: a consensus report from an international seminar on nomenclature and classification, convened by the
WHO collaborating centre for oral cancer. Oral Dis. 2021;27(8):1862–80.
5. Alabi RO, Elmusrati M, Sawazaki-Calone I, Kowalski LP, Haglund C, Coletta RD, Leivo I, et al. Comparison of supervised
machine learning classification techniques in prediction of locoregional recurrences in early oral tongue cancer. Int J Med
Inform. 2020;136: 104068.
6. Müller S, Tilakaratne WM. Update from the 5th edition of the World Health Organization classification of head and neck
tumors: tumours of the oral cavity and mobile tongue. Head Neck Pathol. 2022;16(1):54–62.
7. Shamim MZM, Syed S, Shiblee M, Usman M, Ali SJ, Hussein HS, Farrag M. Automated detection of oral pre-cancerous
tongue lesions using deep learning for early diagnosis of oral cavity cancer. Comput J. 2022;65(1):91–104.
8. Ansarin M, Bruschini R, Navach V, Giugliano G, Calabrese L, Chiesa F, Shah JP, et al. Classification of glossectomies: proposal
for tongue cancer resections. Head Neck. 2019;41(3):821–7.
9. Bansal S, Jadon RS, Gupta SK. Lips and tongue cancer classification using deep learning neural network. In: 2023 6th
international conference on information systems and computer networks (ISCON). IEEE; 2023, pp 1–3.
10. Dwivedi K, Patel K, Pandey JP, Garg P. An automatic robust deep learning and feature fusion-based classification method
for early diagnosis of oral cancer using lip and tongue images. In: 2024 2nd international conference on disruptive tech-
nologies (ICDT). IEEE; 2024, pp. 391–395.
11. Rajaguru H, Prabhakar SK. Performance comparison of oral cancer classification with Gaussian mixture measures and multi
layer perceptron. In: The 16th international conference on biomedical engineering: ICBME 2016. Singapore: Springer;
2017. p. 123–9.
12. Müller S. Update from the 4th edition of the World Health Organization of head and neck tumours: tumours of the oral
cavity and mobile tongue. Head Neck Pathol. 2017;11:33–40.
13. Zini A, Czerninski R, Sgan-Cohen HD. Oral cancer over four decades: epidemiology, trends, histology, and survival by
anatomical sites. J Oral Pathol Med. 2010;39(4):299–305.
14. Bagan J, Sarrion G, Jimenez Y. Oral cancer: clinical features. Oral Oncol. 2010;46(6):414–7.
15. Speight PM, Farthing PM. The pathology of oral cancer. Br Dent J. 2018;225(9):841–7.
16. Sciubba JJ. Oral cancer: the importance of early diagnosis and treatment. Am J Clin Dermatol. 2001;2:239–51.
17. Welikala RA, Remagnino P, Lim JH, Chan CS, Rajendran S, Kallarakkal TG, Yap MH, et al. Automated detection and classifica-
tion of oral lesions using deep learning for early detection of oral cancer. IEEE Access. 2020;8:132677–93.
18. Panigrahi S, Nanda BS, Bhuyan R, et al. Classifying histopathological images of oral squamous cell carcinoma using deep
transfer learning. Heliyon. 2023;9:e13444.
19. Smirnov EA, Timoshenko DM, Andrianov SN. Comparison of regularization methods for ImageNet classification with deep
convolutional neural networks. AASRI Procedia. 2014;6:89–94.
20. Krizhevsky A, Sutskever I, Hinton GE. Imagenet classification with deep convolutional neural networks. Commun ACM.
2017;60(6):84–90.
21. Szegedy C, Liu W, Jia Y, Sermanet P, Reed S, Anguelov D, Rabinovich A et al. Going deeper with convolutions. In: Proceed-
ings of the IEEE conference on computer vision and pattern recognition. 2015, pp. 1–9.
22. Chollet F. Xception: deep learning with depthwise separable convolutions. In: Proceedings of the IEEE conference on
computer vision and pattern recognition. 2017, pp. 1251–1258.
23. Simonyan K, Zisserman A. Very deep convolutional networks for large-scale image recognition. arXiv preprint
arXiv:1409.1556. 2014.
24. He K, Zhang X, Ren S, Sun J. Deep residual learning for image recognition. In: Proceedings of the IEEE conference on
computer vision and pattern recognition. 2016, pp. 770–778.
25. Howard AG, Zhu M, Chen B, Kalenichenko D, Wang W, Weyand T, Adam H et al. MobileNets: efficient convolutional neural
networks for mobile vision applications. arXiv preprint arXiv:1704.04861. 2017.
26. Litjens G, Kooi T, Bejnordi BE, Setio AAA, Ciompi F, Ghafoorian M, Sánchez CI, et al. A survey on deep learning in medical
image analysis. Med Image Anal. 2017;42:60–88.
27. Roth HR, Lu L, Liu J, Yao J, Seff A, Cherry K, Summers RM, et al. Improving computer-aided detection using convolutional
neural networks and random view aggregation. IEEE Trans Med Imaging. 2016;35(5):1170–81.
28. Shin HC, Roth HR, Gao M, Lu L, Xu Z, Nogues I, Summers RM, et al. Deep convolutional neural networks for com-
puter-aided detection: CNN architectures, dataset characteristics and transfer learning. IEEE Trans Med Imaging.
2016;35(5):1285–98.
29. Szegedy C, Vanhoucke V, Ioffe S, Shlens J, Wojna Z. Rethinking the Inception architecture for computer vision. In: Proceed-
ings of the IEEE conference on computer vision and pattern recognition. 2016, pp. 2818–2826.
30. Ker J, Wang L, Rao J, Lim T. Deep learning applications in medical image analysis. IEEE Access. 2017;6:9375–89.
31. Mehmood A, Iqbal M, Mehmood Z, Irtaza A, Nawaz M, Nazir T, Masood M. Prediction of heart disease using deep convolu-
tional neural networks. Arab J Sci Eng. 2020;46:1–14.
32. Shamim MZM, Syed S, Shiblee M, Usman M, Ali S. Automated detection of oral pre-cancerous tongue lesions using deep
learning for early diagnosis of oral cavity cancer. arXiv preprint arXiv:1909.08987. 2019.
33. Kim DW, Lee S, Kwon S, Nam W, Cha IH, Kim HJ. Deep learning-based survival prediction of oral cancer patients. Sci Rep.
2019;9(1):1–10.
34. Mohd F, Noor N, Bakar ZA, Rajion ZA. Analysis of oral cancer prediction using features selection with machine learning. In:
The 7th international conference on information technology (ICIT 2015). 2015, pp. 283–288.
35. Krishnan MMR, Chakraborty C, Ray AK. Wavelet based texture classification of oral histopathological sections. Int J Microsc
Sci Technol Appl Educ. 2010;2(4):897–906.
36. Krishnan MMR, Acharya U, Chakraborty C, Ray A. Automated diagnosis of oral cancer using higher order spectra features
and local binary pattern: a comparative study. Technol Cancer Res Treat. 2011;10(5):443–55.
Kabir et al. Discover Artificial Intelligence (2025) 5:173 Page 21 of 21

37. Patra R, Chakraborty C, Chatterjee J. Textural analysis of spinous layer for grading oral submucous fibrosis. Int J Comput
Appl. 2012;47:975–8887.
38. Krishnan MMR, Shah P, Choudhary A, Chakraborty C, Paul RR, Ray AK. Textural characterization of histopathological images
for oral sub-mucous fibrosis detection. Tissue Cell. 2011;43(5):318–30.
39. Rahman T, Mahanta L, Chakraborty C, Das A, Sarma J. Textural pattern classification for oral squamous cell carcinoma. J
Microsc. 2018;269(1):85–93.
40. Rahman TY, Mahanta LB, Das AK, Sarma JD. Automated oral squamous cell carcinoma identification using shape, texture
and color features of whole image strips. Tissue Cell. 2020;63: 101322.
41. Rahman TY. A histopathological image repository of normal epithelium of oral cavity and oral squamous cell carcinoma.
Mendeley Data. 2019. [Link]
42. Reinhard E, Adhikhmin M, Gooch B, Shirley P. Color transfer between images. IEEE Comput Graph Appl. 2001;21(5):34–41.
43. Shih FY. Image processing and pattern recognition: fundamentals and techniques. New Jersey: Wiley; 2010.
44. Krishnan MMR, Chakraborty C, Paul RR, Ray AK. Hybrid segmentation, characterization and classification of basal cell nuclei
from histopathological images of normal oral mucosa and oral submucous fibrosis. Expert Syst Appl. 2012;39(1):1062–77.
45. Mikołajczyk A, Grochowski M. Data augmentation for improving deep learning in image classification problem. In: 2018
international interdisciplinary PhD workshop (IIPhDW). IEEE; 2018, pp. 117–122.
46. Nair V, Hinton GE. Rectified linear units improve restricted Boltzmann machines. In: Proceedings of the 27th international
conference on machine learning (ICML). 2010.
47. Srivastava N, Hinton G, Krizhevsky A, Sutskever I, Salakhutdinov R. Dropout: a simple way to prevent neural networks from
overfitting. J Mach Learn Res. 2014;15(1):1929–58.
48. Arslan S, Kaya MK, Tasci B, Kaya S, Tasci G, Ozsoy F, Dogan S, Tuncer T. Attention turkernext: investigations into bipolar
disorder detection using OCT images. Diagnostics. 2023;13:3422. [Link]
49. Smith MQP, Ruxton GD. Effective use of the McNemar test. Behav Ecol Sociobiol. 2020;74(11):1–9.
50. Thomas B, Kumar V, Saini S. Texture analysis based segmentation and classification of oral cancer lesions in color images
using ANN. In: 2013 IEEE international conference on signal processing, computing and control (ISPCC). IEEE; 2013, pp.
1–5.
51. Coşgun Baybars S, Talu MH, Danacı Ç, Tuncer SA. Artificial intelligence in oral diagnosis: detecting coated tongue with
convolutional neural networks. Diagnostics. 2025;15(8):1024.
52. Dwivedi K, Chugh B, Srivastava A, Pandey JP. Transens-network: an optimized light-weight transformer and feature fusion
based approach of deep learning models for the classification of oral cancer. Int J Comput Model Appl. 2024;1(1):32–44.
53. Sampath P, Sasikaladevi N, Vimal S, Kaliappan M. Oralnet: deep learning fusion for oral cancer identification from lips and
tongue images using stochastic gradient based logistic regression. Netw Model Anal Health Inform Bioinform. 2024.
[Link]
54. Bansal S, Jadon RS, Gupta SK. Lips and tongue cancer classification using deep learning neural networks. Gwalior: Depart-
ment of Computer Science & Applications Jiwaji University; 2023.

Publisher's Note
Springer Nature remains neutral with regard to jurisdictional claims in published maps and institutional affiliations.

You might also like