D-GRIL: End-to-End Topological Learning with 2-parameter Persistence

Soham Mukherjee, Shreyas N. Samaga, Cheng Xin, Steve Oudot, Tamal K. Dey

arXiv:2406.07100·cs.LG·Published 2024-06-11·Updated 2025-02-22

End-to-end topological learning using 1-parameter persistence is well-known. We show that the framework can be enhanced using 2-parameter persistence by adopting a recently introduced 2-parameter persistence based vectorization technique called GRIL. We establish a theoretical foundation of differentiating GRIL producing D-GRIL. We show that D-GRIL can be used to learn a bifiltration function on standard benchmark graph datasets. Further, we exhibit that this framework can be applied in the context of bio-activity prediction in drug discovery.

TopicsProperty Prediction & ADMET

Tagsdrug-discovery

arXiv categoriescs.LG, cs.AI, math.AT

arXiv abstract pagePDF