В последний раз я проверял компонент построения диаграмм MindFusion для WPF, он поддерживал MVVM посредством привязки данных. Если вы привяжете диаграмму к коллекции ObservableCollection, вы сможете написать собственные шаблоны узлов, которые в основном представляют собой DataTemplate, которые будут отображать элементы YourDiagramModel на панели диаграммы.
В отличие от WinForms, вам обычно не нужно вручную обновлять или рисовать поверхность элемента управления в пользовательских элементах управления WPF.