Skip to Main content Skip to Navigation
Journal articles

A Minimal Model for Classification of Rotated Objects with Prediction of the Angle of Rotation

Abstract : In classification tasks, the robustness against various image transformations remains a crucial property of the Convolutional Neural Networks (CNNs). It can be acquired using the data augmentation. It comes, however, at the price of the risk of overfitting and a considerable increase in training time. Consequently, other ways to endow CNN with invariance to various transformations-and mainly to the rotations-is an intensive field of study. This paper presents a new reduced rotation invariant classification model composed of two parts: a feature representation mapping and a classifier. We provide an insight into the principle and we prove that the proposed model is trainable. This model is smaller in terms of trainable parameters than similar approaches, and has angular prediction capabilities. We illustrate the results on the MNIST and CIFAR-10 datasets. On MNIST, we i) achieve the state of the art of classification on MNIST-rot (with training on MNIST-rot), and ii) improve the results of classification on MNIST-rot (with training on upright MNIST). When trained on CIFAR-10 with upright samples and tested with rotated samples we improve by 20% the state of the art classification results. In all cases, we can predict the rotation angle.
Complete list of metadata

https://hal-mines-paristech.archives-ouvertes.fr/hal-03118567
Contributor : Petr Dokladal Connect in order to contact the contributor
Submitted on : Thursday, February 18, 2021 - 12:05:41 PM
Last modification on : Friday, June 18, 2021 - 4:02:02 PM

File

JVCIR_2021.pdf
Files produced by the author(s)

Identifiers

Citation

Rosemberg Rodriguez Salas, Petr Dokládal, Eva Dokladalova. A Minimal Model for Classification of Rotated Objects with Prediction of the Angle of Rotation. Journal of Visual Communication and Image Representation, Elsevier, 2021, 75, pp.103054. ⟨10.1016/j.jvcir.2021.103054⟩. ⟨hal-03118567v2⟩

Share

Metrics

Record views

1318

Files downloads

300