Add stub dropout probability parameters