arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2005.08235cs.CV

FuCiTNet: Improving the generalization of deep learning networks by the fusion of learned class-inherent transformations

University of Vigo(维戈大学);University of Alicante(阿利坎特大学);University of Granada(格拉纳达大学);Universitat Politècnica de Catalunya(加泰罗尼亚理工大学)

Manuel Rey-Area, Emilio Guirado, Siham Tabik, Javier Ruiz-Hidalgo

英文摘要:

It is widely known that very small datasets produce overfitting in Deep Neural Networks (DNNs), i.e., the network becomes highly biased to the data it has been trained on. This issue is often alleviated using transfer learning, regularization techniques and/or data augmentation. This work presents a new approach, independent but complementary to the previous mentioned techniques, for improving the generalization of DNNs on very small datasets in which the involved classes share many visual features. The proposed methodology, called FuCiTNet (Fusion Class inherent Transformations Network), inspired by GANs, creates as many generators as classes in the problem. Each generator, $k$, learns the transformations that bring the input image into the k-class domain. We introduce a classification loss in the generators to drive the leaning of specific k-class transformations. Our experiments demonstrate that the proposed transformations improve the generalization of the classification model in three diverse datasets.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑