ReGrAt: Regularization in graphs using attention mechanism to handle class imbalance
Published in GCLR workshop, AAAI, 2023
- Neeraja Kirtane, Jeshuren Chelladurai, Balaraman Ravindran, Ashish Tendulkar
Used attention mechanism to tackle imbalance. Used a custom loss function by adding a regularizer that handles imbalance. Got better results than already existing methods.