International Conference on Information and Knowledge Technology

فارسی

Home / سیزدهمین کنفرانس بین المللی فناوری اطلاعات و دانش

Multi-label Classification of Steel Surface Defects Using Transfer Learning and Vision Transformer

Authors :

Amirhossein Komijani¹ Farzaneh Vafaeinezhad² Javad Khoramdel³ Yasamin Borhani⁴ Esmaeil Najafi⁵

1- دانشگاه صنعتی خواجه نصیرالدین طوسی 2- دانشگاه صنعتی خواجه نصیرالدین طوسی 3- دانشگاه تربیت مدرس 4- دانشگاه صنعتی خواجه نصیرالدین طوسی 5- دانشگاه صنعتی خواجه نصیرالدین طوسی

Keywords :

Computer vision،multi-label classification،steel defect detection،deep learning

Abstract :

Steel is the most widely used metal in various industries, for instance, automotive, buildings, packaging, etc. Defect detection on steel surfaces is crucial for steel production companies. This paper proposes a multi-label classification algorithm based on deep learning methods. Several models were trained based on the Severstal dataset, namely MobileNet- V2, Xception, DenseNet121, ViT, and ResNet-50 with transfer learning. In addition, a base model is trained from scratch based on convolutional layers such as ResNet. Moreover, a ViT model is constructed according to the attention and transformer layers concept. At first, 91% for the weighted F1 score is obtained. By exploiting the weighted loss method, the weighted F1 score increased to 92%.

List of archived papers

طراحی واسط کاربری مبتنی بر رفتار و احساسات کاربران در سیستم های هوشمند

فاطمه صبائی - دکتر احمد عبداله زاده بارفروش

A Joint Trajectory and Energy Harvesting Method for an UAV Enabled Disaster Response Network

Hosein Mohammadi Firozjae - Javad Zeraatkar Moghaddam - Mehrdad Ardebilipour

Conceptual Intelligent Model for Visual Question Answering using Attention Mechanism and Relational Reasoning

ٍElham Alighardash - Dr Hassan Khotanlou - Vahid Pour Amin

بررسی روش‌ها، مجموعه‌های داده و معیارهای ارزیابی در حوزه‌ی پرسش از متون درون تصویر

کبری فرشیدی - حسن ختن‌لو - محرم منصوری زاده - الهام علی قارداش

Non-Linear Control of Cancer Model, Considering the Drug Resistance Using Feedback Based Chemotherapy Approach

Danial Kiaei - Hami Tourajizadeh

پیش بینی گره های رهبر در شبکه های اجتماعی با استفاده از پیش بینی پیوند

روح اله رشیدی - فرساد زمانی بروجنی - محمد رضا سلطان آقایی - هادی فرهادی

A Multi-Task Framework Using Mamba for Identity, Age, and Gender Classification from Hand Images

Amirabbas Rezasoltani - Alireza Hosseini - Ramin Toosi - MohammadAli Akhaee

سیستم پیشنهاددهنده غذای سالم با استفاده از داده کاوی عادت های تغذیه ای کاربران

محمد عباسی - مریم حسینی پزوه - محمدرضا شمس

Improving hypergraph attention and hypergraph convolution networks

Mustafa Mohammadi Gharasuie - Mahmood Shabankhah - Ali Kamandi

شکل‌دهی سه بعدی پرتو و بهبود نرخ امن در شبکه‌های مخابراتی بی‌سیم-توان‎‌داده‌شده مبتنی بر صفحات بازتابی هوشمند

کوثر انصاری - دکتر مهدی مجیدی

Samin Hamayesh - Version 40.3.1