Edges are decoded by computing the minimum spanning tree on the weight matrix (the lowest sum of edge distances).
this effectively redistributes probability mass to all other layers, forcing the network to incorporate the information content of all BERt layers.