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

An Improved Drone Detection Method Using Deep Learning for Augmentation Detection Speed

Mohammad Bahrami - Seyyed Amir Asghari - Mohammadreza Binesh Marvasti - Sajjad Ansaria

Extending Interaction Flow Modeling Language as a Profile for Form-making Systems

Ghazaleh Shahin - Dr Bahman Zamani

Predicting Concentration of Particulate Matter (PM2.5) in Hamedan using Machine Learning Algorithms

Anita Karim Ghassabpour - Hatam Abdoli - Muharram Mansoorizadeh - Saeid Seyedi

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

Amirhossein Komijani - Farzaneh Vafaeinezhad - Javad Khoramdel - Yasamin Borhani - Esmaeil Najafi

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

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

جانمایی توزیع‌شده محتوا برای ذخیره‌سازی موقت در شبکه‌های سلولی کوچک با حضور کاربران مخرب

زهرا رشیدی - دکتر وصال حکمی - حانیه سلمان‌طاهری زهرا رشیدی - وصال حکمی - حانیه سلمان‌طاهری -

بررسی کارآمدی فناوری وب 0.2 در پشتیبانی از فرآیندهای انسان محور و دانش مبنا

سید احسان ملیحی - فاطمه مشایخی کردکلا

یک روش خوشه بندی گره ها برای شبکه های حسگر بی‌سیم با هدف بهبود متوازن سازی بار مبتنی بر تکنیک تاپسیس

راضیه حسین رضایی - فهیمه یزدان پناه

Customer Churn Prediction Using Data Mining Techniques for an Iranian Payment Application

Olya Rezaeian - Dr ُSeyedhamidreza Shahabi Haghighi - Dr Jamal Shahrabi

A New Method Based on Deep Learning and Time Stabilization of the Propagation Path for Fake News Detection

Fatemeh Torgheh - Dr Mohammad Reza Keyvanpour - Dr Behrooz Masoumi

more

Samin Hamayesh - Version 42.0.3