Phân loại ImageNet với Mạng nơ-ron Tích chập Sâu
Alex Krizhevsky University of Toronto kriz@cs.utoronto.ca
Ilya Sutskever University of Toronto ilya@cs.utoronto.ca
Geoffrey E. Hinton University of Toronto hinton@cs.utoronto.ca
Tóm tắt
Chúng tôi đã huấn luyện một mạng nơ-ron tích chập sâu, lớn để phân loại 1.2 triệu ảnh độ phân giải cao trong cuộc thi ImageNet LSVRC-2010 thành 1000 lớp khác nhau. Trên dữ liệu kiểm thử, chúng tôi đạt được tỷ lệ lỗi top-1 và top-5 lần lượt là 37.5% và 17.0%, tốt hơn đáng kể so với trạng thái hiện đại trước đó. Mạng nơ-ron, có 60 triệu tham số và 650,000 nơ-ron, bao gồm năm lớp tích chập, một số lớp trong đó được theo sau bởi các lớp max-pooling, và ba lớp kết nối đầy đủ với softmax 1000 chiều cuối cùng. Để làm cho quá trình huấn luyện nhanh hơn, chúng tôi sử dụng các nơ-ron không bão hòa và một triển khai GPU rất hiệu quả của phép toán tích chập. Để giảm hiện tượng quá khớp trong các lớp kết nối đầy đủ, chúng tôi sử dụng một phương pháp chính quy hóa được phát triển gần đây có tên là “dropout”, phương pháp này đã chứng minh là rất hiệu quả. Chúng tôi cũng đưa một biến thể của mô hình này tham gia cuộc thi ILSVRC-2012 và đạt được tỷ lệ lỗi kiểm thử top-5 chiến thắng là 15.3%, so với 26.2% đạt được bởi bài dự thi đứng thứ hai.