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

Optimal control of robotic hand for rehabilitation using fractional order systems and EEG signal processing

Mehran Safari Dehnavi - Vahid Safari Dehnavi - Masoud Shafiee

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

میثم فتاحی

بررسی روش یادگیری انتقالی جهت پیش‌بینی پیوند

علی روحانی فر - کمال میرزایی بدرآبادی

Persian deaf sign language recognition system using deep learning

Mohammad Ebrahimi

مکان‌یابی خطاهای کاربردها و خدمات نرم‌افزاری با کمک تولید داده آزمون با نامتغیرهای محتمل

محمد نصرتی مقدم - حسن حقیقی - مجتبی وحیدی اصل

Targeted Vaccination for COVID-19 Using Mobile Communication Networks

Mohammadmohsen Jadidi - Pegah Moslemi - Saeed Jamshidiha - Iman Masroori - Abbas Mohammadi - Vahid Pourahmadi

A New Sentence Ordering Method Using BERT Pretrained Model

Melika Golestanipour - Seyedeh Zahra Razavi - Dr Heshaam Faili

طراحی و پیاده سازی بستر اجرای بازی جنگ سایبری

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

جایگزینی دارو براساس پیش‌بینی یال روی گراف‌های ناهمگون با بهره‌گیری از جاسازی گراف ناهمگون

رسول سامانی - فهیمه شاهرخ شهرکی - دکتر ناصر قدیری رسول سامانی - فهیمه شاهرخ شهرکی - ناصر قدیری -

A Graph Attention-Based Autoencoder for Critical Path Anomaly Detection in Microservices

Mahdi Naderi - Hossein Momeni - Shayan Shahini

Samin Hamayesh - Version 40.3.1