diff --git a/x-pack/plugins/ml/public/application/components/data_grid/common.ts b/x-pack/plugins/ml/public/application/components/data_grid/common.ts
index f169c56205e08..2805a28996acc 100644
--- a/x-pack/plugins/ml/public/application/components/data_grid/common.ts
+++ b/x-pack/plugins/ml/public/application/components/data_grid/common.ts
@@ -27,7 +27,11 @@ import {
 
 import { DEFAULT_RESULTS_FIELD } from '../../../../common/constants/data_frame_analytics';
 import { extractErrorMessage } from '../../../../common/util/errors';
-import { FeatureImportance, TopClasses } from '../../../../common/types/feature_importance';
+import {
+  FeatureImportance,
+  FeatureImportanceClassName,
+  TopClasses,
+} from '../../../../common/types/feature_importance';
 
 import {
   BASIC_NUMERICAL_TYPES,
@@ -168,8 +172,9 @@ const getClassName = (className: string, isClassTypeBoolean: boolean) => {
 
   return className;
 };
+
 /**
- * Helper to transform feature importance flattened fields with arrays back to object structure
+ * Helper to transform feature importance fields with arrays back to primitive value
  *
  * @param row - EUI data grid data row
  * @param mlResultsField - Data frame analytics results field
@@ -180,69 +185,44 @@ export const getFeatureImportance = (
   mlResultsField: string,
   isClassTypeBoolean = false
 ): FeatureImportance[] => {
-  const featureNames: string[] | undefined =
-    row[`${mlResultsField}.feature_importance.feature_name`];
-  const classNames: string[] | undefined =
-    row[`${mlResultsField}.feature_importance.classes.class_name`];
-  const classImportance: number[] | undefined =
-    row[`${mlResultsField}.feature_importance.classes.importance`];
-
-  if (featureNames === undefined) {
-    return [];
-  }
-
-  // return object structure for classification job
-  if (classNames !== undefined && classImportance !== undefined) {
-    const overallClassNames = classNames?.slice(0, classNames.length / featureNames.length);
-
-    return featureNames.map((fName, index) => {
-      const offset = overallClassNames.length * index;
-      const featureClassImportance = classImportance.slice(
-        offset,
-        offset + overallClassNames.length
-      );
-      return {
-        feature_name: fName,
-        classes: overallClassNames.map((fClassName, fIndex) => {
+  const featureImportance: Array<{
+    feature_name: string[];
+    classes?: Array<{ class_name: FeatureImportanceClassName[]; importance: number[] }>;
+    importance?: number | number[];
+  }> = row[`${mlResultsField}.feature_importance`];
+  if (featureImportance === undefined) return [];
+
+  return featureImportance.map((fi) => ({
+    feature_name: Array.isArray(fi.feature_name) ? fi.feature_name[0] : fi.feature_name,
+    classes: Array.isArray(fi.classes)
+      ? fi.classes.map((c) => {
+          const processedClass = getProcessedFields(c);
           return {
-            class_name: getClassName(fClassName, isClassTypeBoolean),
-            importance: featureClassImportance[fIndex],
+            importance: processedClass.importance,
+            class_name: getClassName(processedClass.class_name, isClassTypeBoolean),
           };
-        }),
-      };
-    });
-  }
-
-  // return object structure for regression job
-  const importance: number[] = row[`${mlResultsField}.feature_importance.importance`];
-  return featureNames.map((fName, index) => ({
-    feature_name: fName,
-    importance: importance[index],
+        })
+      : fi.classes,
+    importance: Array.isArray(fi.importance) ? fi.importance[0] : fi.importance,
   }));
 };
 
 /**
- * Helper to transforms top classes flattened fields with arrays back to object structure
+ * Helper to transforms top classes fields with arrays back to original primitive value
  *
  * @param row - EUI data grid data row
  * @param mlResultsField - Data frame analytics results field
  * @returns nested object structure of feature importance values
  */
 export const getTopClasses = (row: Record<string, any>, mlResultsField: string): TopClasses => {
-  const classNames: string[] | undefined = row[`${mlResultsField}.top_classes.class_name`];
-  const classProbabilities: number[] | undefined =
-    row[`${mlResultsField}.top_classes.class_probability`];
-  const classScores: number[] | undefined = row[`${mlResultsField}.top_classes.class_score`];
-
-  if (classNames === undefined || classProbabilities === undefined || classScores === undefined) {
-    return [];
-  }
-
-  return classNames.map((className, index) => ({
-    class_name: className,
-    class_probability: classProbabilities[index],
-    class_score: classScores[index],
-  }));
+  const topClasses: Array<{
+    class_name: FeatureImportanceClassName[];
+    class_probability: number[];
+    class_score: number[];
+  }> = row[`${mlResultsField}.top_classes`];
+
+  if (topClasses === undefined) return [];
+  return topClasses.map((tc) => getProcessedFields(tc)) as TopClasses;
 };
 
 export const useRenderCellValue = (
diff --git a/x-pack/plugins/ml/public/application/components/data_grid/data_grid.tsx b/x-pack/plugins/ml/public/application/components/data_grid/data_grid.tsx
index da34e0f1bc9fb..5dad9801eb644 100644
--- a/x-pack/plugins/ml/public/application/components/data_grid/data_grid.tsx
+++ b/x-pack/plugins/ml/public/application/components/data_grid/data_grid.tsx
@@ -35,7 +35,7 @@ import {
   getTopClasses,
 } from './common';
 import { UseIndexDataReturnType } from './types';
-import { DecisionPathPopover } from './feature_importance/decision_path_popover';
+import { DecisionPathPopover } from '../../data_frame_analytics/pages/analytics_exploration/components/feature_importance/decision_path_popover';
 import {
   FeatureImportanceBaseline,
   FeatureImportance,
diff --git a/x-pack/plugins/ml/public/application/data_frame_analytics/pages/analytics_exploration/components/expandable_section/expandable_section.scss b/x-pack/plugins/ml/public/application/data_frame_analytics/pages/analytics_exploration/components/expandable_section/expandable_section.scss
index c1c80e8dbd2c4..3a548b40d3a91 100644
--- a/x-pack/plugins/ml/public/application/data_frame_analytics/pages/analytics_exploration/components/expandable_section/expandable_section.scss
+++ b/x-pack/plugins/ml/public/application/data_frame_analytics/pages/analytics_exploration/components/expandable_section/expandable_section.scss
@@ -5,3 +5,9 @@
 .mlExpandableSection-contentPadding {
   padding: $euiSizeS;
 }
+
+// Make sure the charts tooltip in popover
+// have higher zIndex than Eui popover cells
+[id^='echTooltipPortal'] {
+  z-index: $euiZLevel9 !important;
+}
diff --git a/x-pack/plugins/ml/public/application/components/data_grid/feature_importance/decision_path_chart.tsx b/x-pack/plugins/ml/public/application/data_frame_analytics/pages/analytics_exploration/components/feature_importance/decision_path_chart.tsx
similarity index 95%
rename from x-pack/plugins/ml/public/application/components/data_grid/feature_importance/decision_path_chart.tsx
rename to x-pack/plugins/ml/public/application/data_frame_analytics/pages/analytics_exploration/components/feature_importance/decision_path_chart.tsx
index a711d672975aa..5e508df7c6ae5 100644
--- a/x-pack/plugins/ml/public/application/components/data_grid/feature_importance/decision_path_chart.tsx
+++ b/x-pack/plugins/ml/public/application/data_frame_analytics/pages/analytics_exploration/components/feature_importance/decision_path_chart.tsx
@@ -25,12 +25,12 @@ import { EuiIcon } from '@elastic/eui';
 import React, { useCallback, useMemo } from 'react';
 import { i18n } from '@kbn/i18n';
 import euiVars from '@elastic/eui/dist/eui_theme_light.json';
-import { DecisionPathPlotData } from './use_classification_path_data';
-import { formatSingleValue } from '../../../formatters/format_value';
+import type { DecisionPathPlotData } from './use_classification_path_data';
+import { formatSingleValue } from '../../../../../formatters/format_value';
 import {
   FeatureImportanceBaseline,
   isRegressionFeatureImportanceBaseline,
-} from '../../../../../common/types/feature_importance';
+} from '../../../../../../../common/types/feature_importance';
 const { euiColorFullShade, euiColorMediumShade } = euiVars;
 const axisColor = euiColorMediumShade;
 
diff --git a/x-pack/plugins/ml/public/application/components/data_grid/feature_importance/decision_path_classification.tsx b/x-pack/plugins/ml/public/application/data_frame_analytics/pages/analytics_exploration/components/feature_importance/decision_path_classification.tsx
similarity index 97%
rename from x-pack/plugins/ml/public/application/components/data_grid/feature_importance/decision_path_classification.tsx
rename to x-pack/plugins/ml/public/application/data_frame_analytics/pages/analytics_exploration/components/feature_importance/decision_path_classification.tsx
index 48a0c0871f686..d10755b32d7a7 100644
--- a/x-pack/plugins/ml/public/application/components/data_grid/feature_importance/decision_path_classification.tsx
+++ b/x-pack/plugins/ml/public/application/data_frame_analytics/pages/analytics_exploration/components/feature_importance/decision_path_classification.tsx
@@ -14,11 +14,11 @@ import {
   useDecisionPathData,
   getStringBasedClassName,
 } from './use_classification_path_data';
-import {
+import type {
   FeatureImportance,
   FeatureImportanceBaseline,
   TopClasses,
-} from '../../../../../common/types/feature_importance';
+} from '../../../../../../../common/types/feature_importance';
 import { DecisionPathChart } from './decision_path_chart';
 import { MissingDecisionPathCallout } from './missing_decision_path_callout';
 
diff --git a/x-pack/plugins/ml/public/application/components/data_grid/feature_importance/decision_path_json_viewer.tsx b/x-pack/plugins/ml/public/application/data_frame_analytics/pages/analytics_exploration/components/feature_importance/decision_path_json_viewer.tsx
similarity index 86%
rename from x-pack/plugins/ml/public/application/components/data_grid/feature_importance/decision_path_json_viewer.tsx
rename to x-pack/plugins/ml/public/application/data_frame_analytics/pages/analytics_exploration/components/feature_importance/decision_path_json_viewer.tsx
index 93b7bd6bd012f..1110ef8171b96 100644
--- a/x-pack/plugins/ml/public/application/components/data_grid/feature_importance/decision_path_json_viewer.tsx
+++ b/x-pack/plugins/ml/public/application/data_frame_analytics/pages/analytics_exploration/components/feature_importance/decision_path_json_viewer.tsx
@@ -7,7 +7,7 @@
 
 import React, { FC } from 'react';
 import { EuiCodeBlock } from '@elastic/eui';
-import { FeatureImportance } from '../../../../../common/types/feature_importance';
+import type { FeatureImportance } from '../../../../../../../common/types/feature_importance';
 
 interface DecisionPathJSONViewerProps {
   featureImportance: FeatureImportance[];
diff --git a/x-pack/plugins/ml/public/application/components/data_grid/feature_importance/decision_path_popover.tsx b/x-pack/plugins/ml/public/application/data_frame_analytics/pages/analytics_exploration/components/feature_importance/decision_path_popover.tsx
similarity index 94%
rename from x-pack/plugins/ml/public/application/components/data_grid/feature_importance/decision_path_popover.tsx
rename to x-pack/plugins/ml/public/application/data_frame_analytics/pages/analytics_exploration/components/feature_importance/decision_path_popover.tsx
index 3aed0f56d5a76..e1ad6a6863908 100644
--- a/x-pack/plugins/ml/public/application/components/data_grid/feature_importance/decision_path_popover.tsx
+++ b/x-pack/plugins/ml/public/application/data_frame_analytics/pages/analytics_exploration/components/feature_importance/decision_path_popover.tsx
@@ -16,11 +16,11 @@ import {
   isClassificationFeatureImportanceBaseline,
   isRegressionFeatureImportanceBaseline,
   TopClasses,
-} from '../../../../../common/types/feature_importance';
-import { ANALYSIS_CONFIG_TYPE } from '../../../data_frame_analytics/common';
+} from '../../../../../../../common/types/feature_importance';
+import { ANALYSIS_CONFIG_TYPE } from '../../../../common';
 import { ClassificationDecisionPath } from './decision_path_classification';
-import { useMlKibana } from '../../../contexts/kibana';
-import { DataFrameAnalysisConfigType } from '../../../../../common/types/data_frame_analytics';
+import { useMlKibana } from '../../../../../contexts/kibana';
+import type { DataFrameAnalysisConfigType } from '../../../../../../../common/types/data_frame_analytics';
 import { getStringBasedClassName } from './use_classification_path_data';
 
 interface DecisionPathPopoverProps {
diff --git a/x-pack/plugins/ml/public/application/components/data_grid/feature_importance/decision_path_regression.tsx b/x-pack/plugins/ml/public/application/data_frame_analytics/pages/analytics_exploration/components/feature_importance/decision_path_regression.tsx
similarity index 97%
rename from x-pack/plugins/ml/public/application/components/data_grid/feature_importance/decision_path_regression.tsx
rename to x-pack/plugins/ml/public/application/data_frame_analytics/pages/analytics_exploration/components/feature_importance/decision_path_regression.tsx
index ccb7870fd79dc..bb9cdd861788c 100644
--- a/x-pack/plugins/ml/public/application/components/data_grid/feature_importance/decision_path_regression.tsx
+++ b/x-pack/plugins/ml/public/application/data_frame_analytics/pages/analytics_exploration/components/feature_importance/decision_path_regression.tsx
@@ -9,11 +9,11 @@ import React, { FC, useMemo } from 'react';
 import { EuiCallOut } from '@elastic/eui';
 import { FormattedMessage } from '@kbn/i18n/react';
 import d3 from 'd3';
-import {
+import type {
   FeatureImportance,
   FeatureImportanceBaseline,
   TopClasses,
-} from '../../../../../common/types/feature_importance';
+} from '../../../../../../../common/types/feature_importance';
 import { useDecisionPathData, isDecisionPathData } from './use_classification_path_data';
 import { DecisionPathChart } from './decision_path_chart';
 import { MissingDecisionPathCallout } from './missing_decision_path_callout';
diff --git a/x-pack/plugins/ml/public/application/components/data_grid/feature_importance/missing_decision_path_callout.tsx b/x-pack/plugins/ml/public/application/data_frame_analytics/pages/analytics_exploration/components/feature_importance/missing_decision_path_callout.tsx
similarity index 100%
rename from x-pack/plugins/ml/public/application/components/data_grid/feature_importance/missing_decision_path_callout.tsx
rename to x-pack/plugins/ml/public/application/data_frame_analytics/pages/analytics_exploration/components/feature_importance/missing_decision_path_callout.tsx
diff --git a/x-pack/plugins/ml/public/application/components/data_grid/feature_importance/use_classification_path_data.test.tsx b/x-pack/plugins/ml/public/application/data_frame_analytics/pages/analytics_exploration/components/feature_importance/use_classification_path_data.test.tsx
similarity index 98%
rename from x-pack/plugins/ml/public/application/components/data_grid/feature_importance/use_classification_path_data.test.tsx
rename to x-pack/plugins/ml/public/application/data_frame_analytics/pages/analytics_exploration/components/feature_importance/use_classification_path_data.test.tsx
index 18bc02ae63847..70c62294cae00 100644
--- a/x-pack/plugins/ml/public/application/components/data_grid/feature_importance/use_classification_path_data.test.tsx
+++ b/x-pack/plugins/ml/public/application/data_frame_analytics/pages/analytics_exploration/components/feature_importance/use_classification_path_data.test.tsx
@@ -9,7 +9,7 @@ import {
   buildClassificationDecisionPathData,
   buildRegressionDecisionPathData,
 } from './use_classification_path_data';
-import { FeatureImportance } from '../../../../../common/types/feature_importance';
+import type { FeatureImportance } from '../../../../../../../common/types/feature_importance';
 
 describe('buildClassificationDecisionPathData()', () => {
   test('should return correct prediction probability for binary classification', () => {
diff --git a/x-pack/plugins/ml/public/application/components/data_grid/feature_importance/use_classification_path_data.tsx b/x-pack/plugins/ml/public/application/data_frame_analytics/pages/analytics_exploration/components/feature_importance/use_classification_path_data.tsx
similarity index 98%
rename from x-pack/plugins/ml/public/application/components/data_grid/feature_importance/use_classification_path_data.tsx
rename to x-pack/plugins/ml/public/application/data_frame_analytics/pages/analytics_exploration/components/feature_importance/use_classification_path_data.tsx
index ccee43a8c971d..5d61d8b3ef0c4 100644
--- a/x-pack/plugins/ml/public/application/components/data_grid/feature_importance/use_classification_path_data.tsx
+++ b/x-pack/plugins/ml/public/application/data_frame_analytics/pages/analytics_exploration/components/feature_importance/use_classification_path_data.tsx
@@ -14,8 +14,8 @@ import {
   isClassificationFeatureImportanceBaseline,
   isRegressionFeatureImportanceBaseline,
   TopClasses,
-} from '../../../../../common/types/feature_importance';
-import { ExtendedFeatureImportance } from './decision_path_popover';
+} from '../../../../../../../common/types/feature_importance';
+import type { ExtendedFeatureImportance } from './decision_path_popover';
 
 export type DecisionPathPlotData = Array<[string, number, number]>;
 
diff --git a/x-pack/test/functional/apps/ml/data_frame_analytics/feature_importance.ts b/x-pack/test/functional/apps/ml/data_frame_analytics/feature_importance.ts
index b8bdc7de16e1e..49728603c246c 100644
--- a/x-pack/test/functional/apps/ml/data_frame_analytics/feature_importance.ts
+++ b/x-pack/test/functional/apps/ml/data_frame_analytics/feature_importance.ts
@@ -14,8 +14,7 @@ export default function ({ getService }: FtrProviderContext) {
   const esArchiver = getService('esArchiver');
   const ml = getService('ml');
 
-  // Failing: See https://github.com/elastic/kibana/issues/90526
-  describe.skip('total feature importance panel and decision path popover', function () {
+  describe('total feature importance panel and decision path popover', function () {
     const testDataList: Array<{
       suiteTitle: string;
       archive: string;