Farthest-cell balance regularizer
Implementation & benchmark of arXiv:2609.02362 — Farthest-cell triplet entropy: high-dimensional shell limits and hyperbolic curvature amplification
Source paper: Farthest-cell triplet entropy: high-dimensional shell limits and hyperbolic curvature amplification arXiv:2609.02362 ⓘ · analyzed Sep 3, 2026
AI-generated research hypothesis, automatically tested. Not peer-reviewed.
Idea description
Add a regularizer that rewards balanced farthest-prototype cells in an embedding space. For each query and random triple of prototypes, compute which prototype is farthest, estimate the three label probabilities over queries, and maximize their Shannon entropy to discourage prototype domination and representation collapse toward a radial direction.
Formulas
Mathematical statement
For a prototype triple P=(p_1,p_2,p_3) and query random variable X, define the farthest label J_P(X)=argmax_{j in {1,2,3}} d(f(X),p_j), where f(X) is the learned representation and d is the model distance. The farthest-cell probabilities are q_j(P)=Pr_X[J_P(X)=j | P]. The paper's statistic is H_FC=E_P[-sum_{j=1}^3 q_j(P) log q_j(P)], with 0 <= H_FC <= log 3; it is maximal when all three cells have probability 1/3 and near zero when one prototype is farthest for almost every query. For minibatch training, estimate q_j by q_hat_j=(1/B) sum_{b=1}^B w_bj, where w_bj=exp(beta d(z_b,p_j))/sum_{k=1}^3 exp(beta d(z_b,p_k)), z_b=f(x_b), and beta>0 is a soft-farthest inverse temperature. Add L_balance=-E_P[-sum_j q_hat_j log(q_hat_j+epsilon)] to the task loss, so minimizing the total loss maximizes entropy. Hard argmax labels can be used for a detached diagnostic; the soft version supplies gradients.
Implementation notes
Integrate this at the embedding output of a classifier, metric-learning encoder, graph neural network, or retrieval tower, before the task head. In every training step, sample T triples of prototype vectors from the current minibatch or from a maintained queue. For each triple and each embedding z_b, compute the three distances and the soft-farthest weights w_bj above. Average weights over the query minibatch to obtain q_hat_j, compute entropy, and subtract alpha H_FC from the ordinary loss. Use Euclidean distance initially; for normalized embeddings, squared Euclidean distance or cosine-derived distance is sufficient because the hard statistic depends only on ordering. Start with beta equal to 5 divided by the running median pairwise distance, ramp alpha from zero over the first 10% of training, and clip the regularizer so it cannot exceed 10% of the task loss. Keep a detached hard-label metric using j=argmax_j d(z_b,p_j) to verify that the soft surrogate reflects actual farthest-cell balance. The paper supplies the entropy construction and boundedness; minibatch sampling, softening, and weighting are engineering adaptations. First test on CIFAR-10 with a small ResNet-18 and supervised contrastive loss, comparing no regularizer, ordinary nearest-cell entropy, and this farthest-cell entropy at equal compute. Track hard H_FC/log 3, class accuracy, embedding collapse indicators, and robustness under monotone distance rescaling. Success means higher hard farthest entropy without accuracy loss, fewer dominant prototypes, and improved retrieval or corruption performance.
Verification
This idea has not been verified yet.
Verification happens in two stages: Stage 1 — a mechanism check on a toy system confirms the claimed mathematical phenomenon reproduces; Stage 2 — a benchmark implements the idea on a real (small) neural network task and compares it against a tuned baseline over 8 paired seeds with a permutation test.
Artifacts
Artifacts unavailable.