Figure 9 shows that class 5 has the highest correct
predictions, with 2708 samples accurately classified.
This shows that the model performs exceptionally
well for this class. Class 2 also performs well, with
164 correct predictions, indicating good
discrimination for this class.
4.3 Discussion
The CNN model built under Experiment 1 with the
ISIC 2018 dataset achieved an accuracy of 83%
when deployed as a test on the same dataset. The
model's application on the ISIC 2019 dataset,
recorded an accuracy of only 62%. It indicates the
limited ability of the model to generalize the
assigned task in data from different distributions
such as ISIC 2019. Next, the MobileNetV2 model
was trained on the ISIC 2018 dataset, and it
performed better, achieving a high accuracy of 89%
when tested on the test dataset. When tested using
the ISIC 2019 database, this model showed a higher
level of generalization than the CNN, at 72%.
However, the performance gap shows that the model
is not entirely prepared for classifying images from
the 2019 database with training focusing only on the
2018 data. FL method aggregated model weights of
different clients and redistributed the new weights to
each client for refining the local models training. It
proved to be much more effective in building the
models that generalizes well to a wider variety of
datasets other than the ISIC 2018 dataset.
By applying this methodology, we noticed the
improved performance in both models. The local
CNN model with global FedAVG’S performance for
the classification of 2019 images increased to 76%
from 72% whereas for 2018 images the performance
slighted reduced to 82%. However, the model is well
suited for the different datasets. To have an
efficiency in classification, MobileNetV2 was used
as a local client model under the FedAvg algorithm.
This configuration achieved its highest accuracy,
with an 80% for ISIC 2018 and an impressive score
of 87% for the ISIC 2019 dataset. This model is well
suited for both datasets since the model weights are
aggregated globally and the updated weights are sent
back to all the four clients.
5 CONCLUSION AND FUTURE
WORK
In this work, we have designed a skin cancer
classification system using the federated learning
approach with the MobileNetV2 model. This
method has the potential to perform efficient,
privacy-preserving training on decentralized devices
while maintaining high performance. The final
accuracy of the model was 87% for ISIC 2019
dataset and 80% for ISIC 2018 dataset, making it a
promising candidate for use in real-world
applications such as skin cancer detection.
MobileNetV2 is a lightweight architecture and, thus,
more suitable for an edge device with a balance
between performance and computational efficiency.
We ensured that there was no sensitive medical
information leaving the user's device by using
federated learning to preserve privacy. Future work
would involve improving the dataset by applying
SMOTE , model optimization along with federated
learning methods used, and would allow continuous
learning, which makes the system faster on real-time
skin cancer detection on mobile devices. Federated
learning techniques could also be extended to make
this process better in terms of efficiency as well as
accuracy while training models. Such techniques as
Federated Averaging could be combined with even
more advanced techniques such as differential
privacy or secure multi-party computation to further
enhance the privacy and security of the model
updates.
REFERENCES
Sandler, M., A. Howard, M. Zhu, A. Zhmoginov and L.C.
Chen., 2018. Mobilenetv2: Inverted residuals and
linear bottlenecks. Proceedings of the IEEE conference
on computer vision and pattern recognition
Guan, H., P.-T. Yap, A. Bozoki and M. Liu., 2024.
"Federated learning for medical image analysis: A
survey." Pattern Recognition: 110424
Hossen, M. N., V. Panneerselvam, D. Koundal, K. Ahmed,
F. M. Bui and S. M. Ibrahim., 2022. "Federated
machine learning for detection of skin diseases and
enhancement of internet of medical things (IoMT)
security." IEEE journal of biomedical and health
informatics 27(2): 835-841.
Ali, K., Z. A. Shaikh, A. A. Khan and A. A. Laghari.,
2022. "Multiclass skin cancer classification using
EfficientNets–a first step towards preventing skin
cancer." Neuroscience Informatics 2(4): 100034.
Gautam, G. K., S. Singh and A. Singh., 2024. Skin Cancer
Identification Using Deep Learning Technique. 2024
International Conference on Communication,
Computer Sciences and Engineering (IC3SE), IEEE.
Lilhore, U. K., S. Simaiya, Y. K. Sharma, K. S. Kaswan,
K. B. Rao, V. M. Rao, A. Baliyan, A. Bijalwan and R.
Alroobaea., 2024. "A precise model for skin cancer
diagnosis using hybrid U-Net and improved