MHA: Make contribution of children in the tree exponentially decay by depth