Binary Stochastic Representations for Large Multi-class Classification - Archive ouverte HAL
Communication Dans Un Congrès Année : 2017

Binary Stochastic Representations for Large Multi-class Classification

Résumé

Classification with a large number of classes is a key problem in machine learning and corresponds to many real-world applications like tagging of images or textual documents in social networks. If one-vs-all methods usually reach top performance in this context, these approaches suffer of a high inference complexity, linear w.r.t. the number of categories. Different models based on the notion of binary codes have been proposed to overcome this limitation, achieving in a sublinear inference complexity. But they a priori need to decide which binary code to associate to which category before learning using more or less complex heuristics. We propose a new end-to-end model which aims at simultaneously learning to associate binary codes with categories, but also learning to map inputs to binary codes. This approach called Deep Stochastic Neural Codes (DSNC) keeps the sublinear inference complexity but do not need any a priori tuning. Experimental results on different datasets show the effectiveness of the approach w.r.t. baseline methods.

Dates et versions

hal-04003146 , version 1 (23-02-2023)

Identifiants

Citer

Thomas Gerald, Nicolas Baskiotis, Ludovic Denoyer. Binary Stochastic Representations for Large Multi-class Classification. The 24th International Conference on Neural Information Processing, Nov 2017, Guangzhou, China. pp.155-165, ⟨10.1007/978-3-319-70087-8_17⟩. ⟨hal-04003146⟩
64 Consultations
0 Téléchargements

Altmetric

Partager

More