new version according to graph learning paper