使用 PMML 实现模型融合及优化技巧

来源：cnblogs　　作者：AKA栗子　　时间：2024/3/13 9:17:42　　对本文有异议

在机器学习的生产环境中，我们经常需要将多个模型的预测结果进行融合，以便提高预测的准确性。这个过程通常涉及到多个模型子分的简单逻辑回归融合。虽然离线训练时我们可以直接使用sklearn的逻辑回归进行训练和调参，但在生产环境中，模型的上线往往需要使用PMML（Predictive Model Markup Language）格式。PMML不仅能够在一个文件中完成多个模型的融合，还可以输出融合后的Sigmoid打分，极大地方便了生产环境中的模型管理和使用。

使用PMML的Segmentation功能进行模型融合

通过阅读PMML文档，我们发现可以利用其Segmentation功能来实现模型的融合。在PMML文件的最外层使用Segmentation并指定multipleModelMethod为"SelectAll"，这样就可以在最终输出中包含所有内层Segment的输出。这一功能为模型融合提供了便利。

处理单模型分数缺失问题

在实际应用中，我们可能会遇到单个模型分数缺失或等于某个固定常数（例如-999，代表缺失）的情况。这时，我们希望将其视为融合失败，即不进行融合处理，直接返回一个特定的融合失败值（如-999）。为了实现这一逻辑，我们可以使用Segmentation来包裹整个回归过程，设置multipleModelMethod为selectFirst。这样，当内部的第一个Segment检测到某个模型子分等于-999时，就会选择这条路径，并最终输出-999作为融合失败的标志。

<Segmentation multipleModelMethod="selectFirst">
  <Segment id="model1SegmentInvalidInput">
    <CompoundPredicate booleanOperator="or">
      <SimplePredicate field="field1" operator="equal" value="-999"/>
      <SimplePredicate field="field2" operator="equal" value="-999"/>
    </CompoundPredicate>
    <RegressionModel modelName="InvalidInputModelmodel1" functionName="regression" targetFieldName="result">
      <MiningSchema>
        <MiningField name="field1" usageType="active"/>
        <MiningField name="field2" usageType="active"/>
        <MiningField name="result" usageType="predicted"/>
      </MiningSchema>
      <RegressionTable intercept="-999"/>
    </RegressionModel>
  </Segment>
  ...
</Segmentation>

将最终分数转换为概率值

为了将回归后的预测值通过Sigmoid函数转换为概率值，从而和线上使用的分数保持一致的量纲，我们可以在有效的回归Segment中使用normalizationMethod='logit'。这样，输出的分数就会经过Sigmoid转换，转化为概率值。

<RegressionModel functionName="regression" modelName="ValidInputModelmodel2" algorithmName="regression" normalizationMethod="logit" targetFieldName="result">
  ...
</RegressionModel>

为最终输出重新命名

为了方便调用和解析PMML文件，我们需要在文件的最后为输出统一命名。通过指定OutputField的name和对应的segmentId，我们可以清晰地标识每个输出值的来源，使得输出结果更加直观易懂。

<Output>
  <OutputField name="NewScore1" feature="predictedValue" segmentId="model1Segment"/>
  <OutputField name="NewScore2" feature="predictedValue" segmentId="model2Segment"/>
</Output>