The k-nearest 3D attention

[834810269]

I noticed that your project also includes k-nearest neighbor frames for 3D attention. Have you tested whether this has any advantages over epipolar attention in terms of performance? Epipolar attention increases a significant amount of additional VRAM usage.