113 lines
4.9 KiB
Markdown
113 lines
4.9 KiB
Markdown
# 推理接口对材料类型的支持分析与实施方案
|
||
|
||
根据最新的需求，`algorithm_model` 表已支持 `material_type` 字段，用于区分不同材料（如 `Pu`、`U` 等）的专用模型。为了在仿真推理阶段能够正确加载对应的模型，需要对现有的推理数据流进行改造，确保“材料类型”这一关键上下文能够从仿真计算传递到推理服务。
|
||
|
||
## 1. 现状分析 (As-Is)
|
||
|
||
### 1.1 数据流现状
|
||
|
||
1. **SimController**: 调用 `simBuilder.buildUnits` 构建仿真单元。
|
||
2. **SimBuilder**: 从 `Topology` 和 `MaterialService` 读取数据，构建 `SimUnit`。
|
||
- *当前缺失*: `SimUnit` 仅包含 `materialId`，**未包含** `materialType`。
|
||
- *当前缺失*: `Material` 实体类中也未明确发现 `materialType` 字段（仅有 `name`, `concentrations` 等）。
|
||
3. **SimInferService**: 将 `SimContext` 转换为 `DeviceStepInfo`，并按 `deviceType` 分组。
|
||
- *当前缺失*: 分组逻辑未考虑材料类型。
|
||
4. **DeviceInferService**: 接收分组数据，再次按 `algorithmType` 分组，然后调用 `algorithmModelService.getCurrentModelPath(algo, device)`。
|
||
- *当前缺失*: 调用获取模型路径接口时，缺少 `materialType` 参数，导致无法精确定位到特定材料的模型（可能默认返回第一个或报错）。
|
||
|
||
### 1.2 关键断点
|
||
|
||
- **源头缺失**: 仿真单元（SimUnit）不知道自己处理的是什么类型的材料。
|
||
- **传递中断**: 推理服务（InferService）无法获取材料类型信息。
|
||
- **查询不足**: 模型匹配逻辑未利用材料类型进行过滤。
|
||
|
||
## 2. 实施方案 (To-Be)
|
||
|
||
### 2.1 第一步：完善领域模型 (Domain & Model)
|
||
|
||
#### 1. 不修改 `Material` 实体
|
||
|
||
需要根据材料表获取的是U浓度值或者Pu浓度值，决定Material\_type = U 或者Pu
|
||
|
||
这个主要是模型根据材料类型训练来的，所以只能把数据也做了材料分类。
|
||
|
||
#### 2. 修改 `SimUnit` Record
|
||
|
||
在 `SimUnit` 中增加 `materialType` 字段，用于在内存中流转该信息。
|
||
|
||
```java
|
||
public record SimUnit(
|
||
String unitId,
|
||
String deviceId,
|
||
String materialId,
|
||
String deviceType,
|
||
String materialType, // 新增
|
||
Map<String, Double> staticProperties
|
||
) {}
|
||
```
|
||
|
||
#### 3. 修改 `DeviceStepInfo`
|
||
|
||
在 `DeviceStepInfo` 中增加 `materialType` 字段，确保分组时能携带此信息。
|
||
|
||
```java
|
||
public class DeviceStepInfo {
|
||
// ... 原有字段
|
||
private String materialType; // 新增
|
||
|
||
// getter/setter ...
|
||
}
|
||
```
|
||
|
||
### 2.2 第二步：更新构建逻辑 (SimBuilder)
|
||
|
||
在 `SimBuilder.buildUnits` 方法中：
|
||
|
||
1. 在查询 `Material` 信息时，根据材料属性（U浓度/Pu浓度等）推导 `materialType`。
|
||
- 规则示例：若 `pu_concentration > 0` 则为 `Pu`，否则若 `u_concentration > 0` 则为 `U`，否则为 `unknown`。
|
||
2. 在构造 `SimUnit` 时，将 `material_type` 注入。
|
||
|
||
### 2.3 第三步：更新推理准备逻辑 (SimInferService)
|
||
|
||
在 `SimInferService.asyncInferAndSave` 方法中：
|
||
|
||
1. 在将 `SimContext` 转换为 `DeviceStepInfo` 时，从对应的 `SimUnit` 中提取 `materialType` 并赋值给 `DeviceStepInfo`。
|
||
2. **分组策略变更**:
|
||
- 原: `Map<String, List<DeviceStepInfo>>` (Key: DeviceType)
|
||
- 新: 保持外层按 `DeviceType` 分组不变，利用 `DeviceStepInfo` 内部携带的 `materialType` 在下一步进行细分。
|
||
|
||
### 2.4 第四步：更新推理执行逻辑 (DeviceInferService)
|
||
|
||
在 `DeviceInferService.processDeviceInference` 方法中：
|
||
|
||
1. 接收到的数据是按 `DeviceType` 分组的。
|
||
2. **三级分组**: 遍历设备列表，进一步按 `AlgorithmType` **AND** `MaterialType` 进行分组。
|
||
- 构造复合键或嵌套 Map: `Map<String, Map<String, List<DeviceStepInfo>>>` (AlgoType -> MaterialType -> List)。
|
||
3. **模型匹配**:
|
||
```java
|
||
String modelPath = algorithmModelService.getCurrentModelPath(currentAlgoType, deviceType, currentMaterialType);
|
||
```
|
||
4. **请求构建**: 确保找到的模型是与当前批次数据的材料类型匹配的。
|
||
|
||
### 2.5 第五步：更新模型查询接口 (AlgorithmModelService)
|
||
|
||
完善 `getCurrentModelPath` 方法，增加 `materialType` 参数，并更新 MyBatis-Plus 查询条件。
|
||
|
||
```java
|
||
String getCurrentModelPath(String algorithmType, String deviceType, String materialType);
|
||
```
|
||
|
||
## 3. 任务清单 (Action Items)
|
||
|
||
1. **Domain & Model**:
|
||
- [ ] 更新 `SimUnit.java` 增加 `materialType`。
|
||
- [ ] 更新 `DeviceStepInfo.java` 增加 `materialType`。
|
||
2. **Builder**:
|
||
- [ ] 修改 `SimBuilder.java`，实现材料类型推导逻辑并填充 `SimUnit`。
|
||
3. **Service**:
|
||
- [ ] 修改 `SimInferService.java`，在转换时将 `materialType` 传递给 `DeviceStepInfo`。
|
||
- [ ] 修改 `DeviceInferService.java`，实现 (Algo + Material) 的组合分组与模型匹配调用。
|
||
- [ ] 修改 `AlgorithmModelService.java` 及实现类，增加带 `materialType` 的路径查询方法。
|
||
4. **Controller**:
|
||
- [ ] `SimController` 无需修改代码，验证端到端流程。
|