Skip to content

Implement motive decomposition network (MDN)#6

Merged
tarikuamisganaw merged 2 commits intoiCog-Labs-Dev:mainfrom
Samrawitgebremaryam:feature/add-motive-decomposition-network
Apr 30, 2026
Merged

Implement motive decomposition network (MDN)#6
tarikuamisganaw merged 2 commits intoiCog-Labs-Dev:mainfrom
Samrawitgebremaryam:feature/add-motive-decomposition-network

Conversation

@Samrawitgebremaryam
Copy link
Copy Markdown
Collaborator

Summary

This PR implements the Motive Decomposition Network (MDN) for SubRep and adds unit tests for its forward-pass behavior.

The MDN maps context features to two context-conditioned outputs:

  • strictly positive Dirichlet alpha parameters over objective weights
  • bounded support geometry values for the admissible weight region

Changes

  • add MotiveDecompositionNetwork in generator/mdn.py
  • add a shared MLP trunk with separate distribution_head and support_head
  • produce Dirichlet alpha parameters using Softplus + epsilon
  • produce support geometry values using Sigmoid
  • validate input shape with clear ValueErrors

Verification

  • manual forward-pass checks completed for single, batched, and invalid inputs
  • full test suite passes locally with:
python -m pytest -v

@tarikuamisganaw tarikuamisganaw merged commit 6b06408 into iCog-Labs-Dev:main Apr 30, 2026
1 check passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants