TCMP: end-to-end topologically consistent magnitude pruning for miniaturized graph convolutional networks - Archive ouverte HAL
Communication Dans Un Congrès Année : 2024

TCMP: end-to-end topologically consistent magnitude pruning for miniaturized graph convolutional networks

Résumé

Magnitude pruning is one of the mainstream methods in lightweight architecture design whose goal is to extract subnetworks with the largest weight connections. This method is known to be successful, but under very high pruning regimes, it suffers from topological inconsistency which renders the extracted subnetworks disconnected, and this hinders their generalization ability. In this paper, we devise TCMP a novel end-to-end Topologically Consistent Magnitude Pruning method that allows extracting subnetworks while guaranteeing their topological consistency. The latter ensures that only accessible and co-accessible -impactful -connections are kept in the resulting lightweight networks. Our solution is based on a novel reparametrization and two supervisory bi-directional networks which implement accessibility/coaccessibility and guarantee that only connected subnetworks will be selected during training. This solution allows enhancing generalization significantly, under very high pruning regimes, as corroborated through extensive experiments, involving graph convolutional networks, on the challenging task of skeleton-based action recognition.
Fichier principal
Vignette du fichier
document.pdf (287.03 Ko) Télécharger le fichier
Origine Fichiers produits par l'(les) auteur(s)

Dates et versions

hal-04796208 , version 1 (21-11-2024)

Identifiants

Citer

Hichem Sahbi. TCMP: end-to-end topologically consistent magnitude pruning for miniaturized graph convolutional networks. ICASSP 2024 - 2024 IEEE International Conference on Acoustics, Speech and Signal Processing (ICASSP), Apr 2024, Seoul, South Korea. pp.3065-3069, ⟨10.1109/ICASSP48485.2024.10446196⟩. ⟨hal-04796208⟩
0 Consultations
0 Téléchargements

Altmetric

Partager

More