def __init__(self):
super(SecureAddHost, self).__init__()
self.y = None
self.y1 = None
self.y2 = None
self.x2 = None
self.x2_plus_y2 = None
self.transfer_inst = SecureAddExampleTransferVariable()
self.model_param = SecureAddExampleParam()
self.data_output = None
self.model_output = None
def __init__(self):
super(SecureAddGuest, self).__init__()
self.x = None
self.x1 = None
self.x2 = None
self.y1 = None
self.x1_plus_y1 = None
self.data_num = None
self.partition = None
self.seed = None
self.transfer_inst = SecureAddExampleTransferVariable()
self.model_param = SecureAddExampleParam()
self.data_output = None
self.model_output = None
def __init__(self):
self.y = None
self.y1 = None
self.y2 = None
self.x2 = None
self.x2_plus_y2 = None
self.transfer_inst = SecureAddExampleTransferVariable()
self.model_param = SecureAddExampleParam()
class SecureAddGuest(ModelBase):
def __init__(self):
super(SecureAddGuest, self).__init__()
self.x = None
self.x1 = None
self.x2 = None
self.y1 = None
self.x1_plus_y1 = None
self.data_num = None
self.partition = None
self.seed = None
self.transfer_inst = SecureAddExampleTransferVariable()
self.model_param = SecureAddExampleParam()
self.data_output = None
self.model_output = None
def _init_runtime_parameters(self, cpn_input):
self.model_param.update(cpn_input.parameters)
self._init_model()
def _init_model(self):
self.data_num = self.model_param.data_num
self.partition = self.model_param.partition
self.seed = self.model_param.seed
def _init_data(self):
kvs = [(i, 1) for i in range(self.data_num)]
self.x = session.parallelize(kvs, include_key=True, partition=self.partition)
def share(self, x):
first = np.random.uniform(x, -x)
return first, x - first
def secure(self):
x_shares = self.x.mapValues(self.share)
self.x1 = x_shares.mapValues(lambda shares: shares[0])
self.x2 = x_shares.mapValues(lambda shares: shares[1])
def add(self):
self.x1_plus_y1 = self.x1.join(self.y1, lambda x, y: x + y)
guest_sum = self.x1_plus_y1.reduce(lambda x, y: x + y)
return guest_sum
def reconstruct(self, guest_sum, host_sum):
print("host sum is %.4f" % host_sum)
print("guest sum is %.4f" % guest_sum)
secure_sum = host_sum + guest_sum
print("Secure Add Result is %.4f" % secure_sum)
return secure_sum
def sync_share_to_host(self):
self.transfer_inst.guest_share.remote(self.x2,
role="host",
idx=0)
def recv_share_from_host(self):
self.y1 = self.transfer_inst.host_share.get(idx=0)
def recv_host_sum_from_host(self):
host_sum = self.transfer_inst.host_sum.get(idx=0)
return host_sum
def run(self, cpn_input):
LOGGER.info("begin to init parameters of secure add example guest")
self._init_runtime_parameters(cpn_input)
LOGGER.info("begin to make guest data")
self._init_data()
LOGGER.info("split data into two random parts")
self.secure()
LOGGER.info("share one random part data to host")
self.sync_share_to_host()
LOGGER.info("get share of one random part data from host")
self.recv_share_from_host()
LOGGER.info("begin to get sum of guest and host")
guest_sum = self.add()
LOGGER.info("receive host sum from guest")
host_sum = self.recv_host_sum_from_host()
secure_sum = self.reconstruct(guest_sum, host_sum)
assert (np.abs(secure_sum - self.data_num * 2) < 1e-6)
LOGGER.info("success to calculate secure_sum, it is {}".format(secure_sum))
return ComponentOutput(self.save_data(), self.export_model(), self.save_cache())
class SecureAddHost(ModelBase):
def __init__(self):
super(SecureAddHost, self).__init__()
self.y = None
self.y1 = None
self.y2 = None
self.x2 = None
self.x2_plus_y2 = None
self.transfer_inst = SecureAddExampleTransferVariable()
self.model_param = SecureAddExampleParam()
self.data_output = None
self.model_output = None
def _init_runtime_parameters(self, cpn_input):
self.model_param.update(cpn_input.parameters)
self._init_model()
def _init_model(self):
self.data_num = self.model_param.data_num
self.partition = self.model_param.partition
self.seed = self.model_param.seed
def _init_data(self):
kvs = [(i, 1) for i in range(self.data_num)]
self.y = session.parallelize(kvs,
include_key=True,
partition=self.partition)
def share(self, y):
first = np.random.uniform(y, -y)
return first, y - first
def secure(self):
y_shares = self.y.mapValues(self.share)
self.y1 = y_shares.mapValues(lambda shares: shares[0])
self.y2 = y_shares.mapValues(lambda shares: shares[1])
def add(self):
self.x2_plus_y2 = self.y2.join(self.x2, lambda y, x: y + x)
host_sum = self.x2_plus_y2.reduce(lambda x, y: x + y)
return host_sum
def sync_share_to_guest(self):
self.transfer_inst.host_share.remote(self.y1, role="guest", idx=0)
def recv_share_from_guest(self):
self.x2 = self.transfer_inst.guest_share.get(idx=0)
def sync_host_sum_to_guest(self, host_sum):
self.transfer_inst.host_sum.remote(host_sum, role="guest", idx=0)
def run(self, cpn_input):
LOGGER.info("begin to init parameters of secure add example host")
self._init_runtime_parameters(cpn_input)
LOGGER.info("begin to make host data")
self._init_data()
LOGGER.info("split data into two random parts")
self.secure()
LOGGER.info("get share of one random part data from guest")
self.recv_share_from_guest()
LOGGER.info("share one random part data to guest")
self.sync_share_to_guest()
LOGGER.info("begin to get sum of host and guest")
host_sum = self.add()
LOGGER.info("send host sum to guest")
self.sync_host_sum_to_guest(host_sum)
return ComponentOutput(self.save_data(), self.export_model(),
self.save_cache())