Skip to main content
. 2023 Feb 26;15:29. doi: 10.1186/s13321-023-00698-9

Table 1.

Algorithm of ABT-MPNN

Initialization
1)

Given a molecular graph G, generate atom features xv and bond features evw

where vAtomG and wNeighborv; three inter-atomic matrices MAdjacency,MDistance,MCoulomb; molecular descriptors hf

2) for each atom v in molecule G:
3)     for each atom w in molecule Neighborv:
4) hvw0ReLU(WiConcatxv,evw)
Bond Embedding Phase
1) Message-passing iteration:t=1,2,,T
2) while 1tT:
3)     for each atom v in molecule G:
4)         for each atom w in molecule Neighborv:
5) mvwtkNeighborvhkvt-1-hwvt-1
6) bvwtBondAttentionmvwt+mvwt
7) hvwtReLU(hvw0+Whbvwt)
Atom Embedding Phase
1) for each atom v in molecule G:
2) mvReLU(WoConcat(xv,wNeighborvhvwT))
3) hvAtomAttentionmv,MAdjacency,MDistance,MCoulomb+mv
Molecule Embedding Phase
1) hvGhv
2) y^FFN(Concat(h,hf))