Graph Refinement based Tree Extraction using Mean-Field Networks and Graph Neural Networks