GLL: A Differentiable Graph Learning Layer for Neural Networks
The graph learning layer (GLL) replaces the usual projection head and softmax classifier with graph-Laplacian label propagation. Samples in a minibatch become nodes of a similarity graph, allowing predictions to use relationships among examples instead of classifying each representation independently.
Dec 11, 2024
