[Mllab] Machine learning registration