Штучний інтелект

Анотація: The paper discusses the approach to binary image classification with ResNet based deep learning. Its actuality is the automatic image classification process and subject-the use of the methods of data augmentation and in the improvement of the model accuracy-deep neural networks. The goal of this research was to develop and assess the efficiency of using the pre-trained ResNet network for solving the binary classification problem. Such image preprocessing and implementation for the augmentation method and mixed learning were conducted to optimize the classification process. The experimental results discussed point towards integration in preprocessing methods, dynamic image loading, and computational process optimization that drive substantial growth in the quality of classification, which carries great practical value in further explorations in this area.

Ключові слова: image classification, ResNet, deep learning, binary classification, data augmentation

Посилання:

1. Korkut, Ş. G., Kocabaş, H., & Kurban, R. (2024). A Comparative Analysis of Convolutional Neural Network Architectures for Binary Image Classification: A Case Study in Skin Cancer Detection. Karadeniz Fen Bilimleri Dergisi, 14(4), 2008–2022. https://doi.org/10.31466/kfbd.1515451
2. Qiu, Y. (2023). Binary-SE-ResNet Model Research for Binary Classification of Chest X-Ray. 462–466. https://doi.org/10.1109/ichci58871.2023.10278004
3. Ye, X., Jin, Y., Wang, Z., Feng, N., Mu, L., & Xie, G. (2023). Image Binary Classification of Coronary Artery Stenosis Based on Resnet18 and Transfer Learning. https://doi.org/10.1109/ccdc58219.2023.10326590
4. Noori, N. M., & Qasim, O. S. (2023). ImageNet Classification using ResNet50 and Binary Marine Predators Algorithm. 340–344. https://doi.org/10.1109/icitams57610.2023.10525313
5. Romeu, E., & Shashev, D. (2024). Binary Convolution Model for Image Classification. https://doi.org/10.1109/itnt60778.2024.10582370
6. Detect AI vs. Human-Generated Images: Dataset. URL:https://www.kaggle.com/competitions/detect-ai-vs-human-generated-images/data
7. PyTorch documentation: RandomResizedCrop. URL:https://pytorch.org/vision/stable/transforms.html#torchvision.transforms.RandomResizedCrop.
8. PyTorch documentation:RandomHorizontalFlip. URL:https://pytorch.org/vision/stable/transforms.html#torchvision.transforms.RandomHorizontalFlip.
9. PyTorch documentation: RandomRotation. URL:https://pytorch.org/vision/stable/transforms.html#torchvision.transforms.RandomRotation.
10. PyTorch documentation: ColourJitter. URL:https://pytorch.org/vision/stable/transforms.html#torchvision.transforms.ColorJitter.
11. PyTorch documentation: ToTensor. URL:https://pytorch.org/vision/stable/transforms.html#torchvision.transforms.ToTensor.
12. PyTorch documentation: Normalise. URL:https://pytorch.org/vision/stable/transforms.html#torchvision.transforms.Normalize.
13. PyTorch documentation: Dataset. URL:https://pytorch.org/docs/stable/data.html#torch.utils.data.Dataset.
14. PyTorch documentation: DataLoader. URL:https://pytorch.org/docs/stable/data.html#torch.utils.data.DataLoader.
15. PyTorch documentation: nn module. URL: https://pytorch.org/docs/stable/nn.html.
16. PyTorch documentation: optim. URL: https://pytorch.org/docs/stable/optim.html.
17. PyTorch documentation: SGD. URL:https://pytorch.org/docs/stable/generated/torch.optim.SGD.html.
18. PyTorch documentation: Adam. URL:https://pytorch.org/docs/stable/generated/torch.optim.Adam.html.
19. PyTorch documentation: amp. URL:https://pytorch.org/docs/stable/amp.html.
20. NN for automatic image classification repository URL:https://github.com/Mihai-MP/NN-for-automatic-image-classification