Effect of Gumbel-softmax temperatures on HSA performance. We show heatmaps of the mean F1-score (left) and standard deviation (right) over multiple runs on the MACHO test set for different combinations of the Gumbel-softmax temperature parameters, τ₁ and τ₂. Warmer colors in the left panel indicate higher mean F1-scores, while cooler colors in the right panel reflect a lower variance.