Skip to Main content Skip to Navigation
Preprints, Working Papers, ...

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.
Document type :
Preprints, Working Papers, ...
Complete list of metadata

https://hal-mines-paristech.archives-ouvertes.fr/hal-03118567
Contributor : Petr Dokladal <>
Submitted on : Friday, January 22, 2021 - 12:44:45 PM
Last modification on : Thursday, February 25, 2021 - 3:28:19 AM

File

yjvci-one-column-format.pdf
Files produced by the author(s)

Identifiers

  • HAL Id : hal-03118567, version 1

Collections

Citation

Rosemberg Rodriguez Salas, Petr Dokládal, Eva Dokladalova. A Minimal Model for Classification of Rotated Objects with Prediction of the Angle of Rotation. 2020. ⟨hal-03118567v1⟩

Share

Metrics

Record views

66

Files downloads

30